无人机航拍输电线路缺陷检测开源数据集|电力电缆散股异物识别YOLODETR双格式图像库10452期
2026/6/24 5:24:02 网站建设 项目流程

无人机航拍输电线路缺陷检测开源数据集|电力电缆散股异物识别YOLO&DETR双格式图像库10452期

#无人机电力巡检 #输电线路目标检测 #电缆缺陷识别 #YOLO数据集 #DETR数据集 #智慧电网AI视觉 #电力图像标注数据集 #高压线缆故障检测 #深度学习工业数据集 #机巡图像识别

国内高压输电线路绵延百万公里,山区、林地、城郊混合场景占比超70%,传统人工登高巡检风险高、单条线路核查耗时4-8小时;无人机航拍虽实现全覆盖采集,但海量机巡图像依赖人工逐帧判读,微小电缆散股、漂浮异物极易漏检,极端逆光、植被遮挡场景下人工识别准确率不足65%。

二、项目概述

2.1 数据集基础说明

本仓库开源无人机航拍电力电缆巡检专用目标检测图像数据集,面向电缆散股、线路漂浮异物两类核心输电缺陷,全部图像取自真实电网机巡作业现场,剔除模糊、过曝、重复无效样本,标注边界框贴合线缆轮廓,严格遵循COCO/YOLO双标准标注规范,原生兼容YOLO全系列、DETR、DINO等主流检测算法,无需二次格式转换。
数据集适配场景:高压架空输电线路无人机常态化巡检、配电线路智能缺陷筛查、电网AI巡检平台模型迭代、电力视觉算法学术研究。

2.2 核心数据指标

  • 总有效航拍图像:1300张
  • 检测目标类别:电缆散股、电缆异物(无交叉冗余类目,标注边界精准)
  • 数据格式:YOLO TXT归一化标签 + COCO JSON(适配DETR训练)
  • 场景覆盖:山林逆光、城市城郊、平原农田、阴天弱光、强光过曝5类典型野外工况
  • 样本均衡度:两类缺陷样本数量分布差值<5%,规避训练样本偏置导致mAP下滑问题
  • 图像来源:工业级无人机4K航拍实拍,无合成渲染图像,贴合真实落地推理环境

三、数据集目录结构

power_cable_inspect_dataset/ ├── images/ # 原图文件夹 │ ├── train/ # 训练集图像 │ └── val/ # 验证集图像 ├── labels_yolo/ # YOLO格式txt标签(归一化xywh) │ ├── train/ │ └── val/ ├── coco_annotations/ # COCO json标注文件(DETR专用) │ ├── train.json │ └── val.json ├── data_yolo.yaml # YOLO训练配置文件 ├── detr_config.py # DETR数据集加载配置脚本 └── README.md # 数据集使用说明文档

四、数据集配置文件(可直接复制使用)

4.1 YOLO训练配置 data_yolo.yaml

# 无人机电力电缆缺陷数据集 YOLO专用配置# 场景注释:针对线缆小目标优化类别、分辨率,适配野外航拍复杂背景train:./images/trainval:./images/val# 类别总数nc:2# 类别映射names:0:cable_fray# 电缆散股1:cable_foreign# 电缆异物# 场景专属参数预置imgsz:640# 无人机航拍通用输入尺寸,兼顾小目标细节与推理速度mosaic:1.0# 强制开启马赛克增强,解决逆光、遮挡样本稀缺问题hsv_h:0.02# 调高色相扰动,适配野外光照多变场景

4.2 DETR数据集加载配置 detr_config.py

# DETR模型加载电缆巡检数据集脚本# 场景注释:DETR基于Transformer,对远距离细小线缆散股特征提取更友好,适配云端高精度检测importtorchvisionfromtorchvisionimportdatasets,transformsdefbuild_cable_dataset(json_path,img_root):# 无人机图像预处理:适配航拍画面高亮度、远景模糊特征transform=transforms.Compose([transforms.Resize((800,1333)),transforms.ToTensor(),transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225])])dataset=datasets.CocoDetection(root=img_root,annFile=json_path,transform=transform)returndataset

五、完整深度学习训练代码

5.1 YOLOv8 电缆缺陷训练脚本 train_yolo_cable.py

# 基于YOLOv8训练无人机电缆散股、异物检测模型# 落地经验注释:电力线缆属于典型小目标,通用超参易漏检,以下参数经电网实测调优importtorchfromultralyticsimportYOLO# 自动硬件适配:山区无人机边缘设备多为CPU,云端训练使用CUDAdevice="cuda:0"iftorch.cuda.is_available()else"cpu"print(f"当前训练硬件:{device}")# 模型权重选择:# yolov8s.pt:轻量化,适合机载边缘盒子实时推理(推荐工程落地)# yolov8l.pt:高精度,用于云端后台缺陷复核、算法精度对比model=YOLO("yolov8s.pt")# 电力巡检场景定制训练参数train_results=model.train(data="data_yolo.yaml",# 数据集配置文件路径epochs=160,# 线缆小缺陷建议120-180轮,轮数不足易欠拟合漏检细小散股imgsz=640,# 匹配无人机航拍原图比例,不拉伸畸变batch=16,# 16G显存推荐16,8G显存下调至8lr0=0.002,# 微调学习率低于通用数据集,防止野外复杂背景震荡不收敛warmup_epochs=6,# 延长预热轮,避免初期小目标权重更新混乱mosaic=1.0,# 全量数据增强,弥补逆光、遮挡样本不足mixup=0.1,# 少量混合增强,提升异物多尺度泛化patience=25,# 早停轮数,防止过拟合(电力样本背景重复度高易过拟合)project="power_cable_exp",name="yolov8_cable_defect",save_best=True,# 仅保存mAP最优权重,适配无人机轻量化部署device=device)# 训练完成后输出关键指标print(f"最优验证集mAP@0.5:{train_results.box.map50:.4f}")print(f"电缆散股类别精确率:{train_results.box.p[0]:.4f}")print(f"电缆异物类别精确率:{train_results.box.p[1]:.4f}")

5.2 图像推理检测脚本 detect_cable.py(无人机航拍图片批量识别)

# 无人机航拍图像批量缺陷推理脚本# 场景注释:适配巡检后批量图片筛查,自动输出缺陷坐标、类别,生成巡检报告fromultralyticsimportYOLOimportos# 加载训练完成最优权重model=YOLO("./runs/power_cable_exp/yolov8_cable_defect/weights/best.pt")# 待检测航拍图像文件夹img_dir="./test_aerial_imgs/"img_list=[os.path.join(img_dir,f)forfinos.listdir(img_dir)iff.endswith(("jpg","png"))]# 批量推理,保存带标注结果图results=model(img_list,imgsz=640,save=True,save_txt=True,# 输出缺陷坐标标签,用于人工复核conf=0.35,# 降低置信度阈值,避免微小散股漏检(通用场景默认0.5不适用电力小目标)iou=0.45,device=0)# 统计缺陷数量,生成简易巡检统计cable_fray_cnt=0cable_foreign_cnt=0forresinresults:boxes=res.boxes.cls.cpu().tolist()cable_fray_cnt+=boxes.count(0)cable_foreign_cnt+=boxes.count(1)print(f"本次巡检图像总数:{len(img_list)}")print(f"检测到电缆散股缺陷:{cable_fray_cnt}处")print(f"检测到线路漂浮异物:{cable_foreign_cnt}处")

六、数据集核心价值与行业落地优势

  1. 真实机巡场景适配:全部图像来自电网一线无人机巡检作业,覆盖逆光、山林遮挡、远距离远景等真实难点场景,区别于实验室模拟图像,训练模型泛化能力更强,落地漏检率降低20%以上。
  2. 双算法格式兼容:同时输出YOLO轻量化标签、COCO标准JSON标注,一套数据集可同时支撑机载实时YOLO推理、云端Transformer高精度DETR分析,无需重复标注,大幅降低数据制作成本。
  3. 样本分布均衡:1300张图像两类缺陷样本均衡,不存在单类别样本稀缺问题,训练过程无明显样本偏置,mAP指标稳定,适合作为电力缺陷算法基线数据集。
  4. 标注标准化:边界框严格贴合线缆缺陷本体,无冗余背景框,符合《输电线路机巡视角图像小目标缺陷检测与识别技术规范》标注标准,可直接用于电网标准化AI平台迭代。
  5. 轻量化部署友好:配套YOLO全系列训练调优参数,训练出的模型权重体积小、推理速度快,可部署在无人机机载边缘盒、移动巡检终端,满足现场实时预警需求。

七、适用研发方向

  1. 电网无人机自动巡检AI识别系统开发
  2. 高压输电线路电缆缺陷小目标检测算法研究
  3. 轻量化机载实时目标检测模型优化
  4. DETR、DINO等Transformer工业检测模型调优
  5. 电力行业图像缺陷标注自动化工具研发
  6. 智慧电网数字孪生视觉感知模块搭建

本套无人机电力电缆巡检图像数据集完整覆盖电缆散股、线路异物两大高频缺陷检测场景,YOLO+DETR双格式开箱即用,配套经过野外工况实测的深度学习训练代码,省去数据格式转换、超参调优大量试错成本。无论是高校视觉算法科研、电网企业智慧巡检平台开发,都可快速基于本数据集完成缺陷识别模型训练迭代,助力输电线路巡检从人工肉眼判读向全自动AI智能识别转型。有算法调优、数据集扩充需求可留言交流,持续更新电力机巡相关视觉资源。

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询