山南市网站建设_网站建设公司_数据统计_seo优化

电网输电线散股检测数据集

4000+。
支持yolo和voc格式。
已划分好训练集测试集验证集
map0.85以上

1

1

以下是电网输电线散股检测数据集的完整说明与YOLOv8 训练代码，支持：

✅ 数据集：4000+ 张图像
✅ 格式：YOLO（.txt） + VOC（.xml）双格式
✅ 已划分：训练集、验证集、测试集
✅ mAP@0.5 > 0.85（高精度）
✅ 提供完整 YOLOv8 训练流程（含data.yaml、train.py、可视化）

✅ 一、数据集概览

📂 数据结构示例：

power_line_dataset/ ├── train/ │ ├── images/ │ └── labels/ ├── val/ │ ├── images/ │ └── labels/ └── test/ ├── images/ └── labels/

✅ 二、标注格式说明

1. YOLO 格式（.txt）

每行：class x_center y_center width height（归一化）

0 0.45 0.32 0.12 0.08

• 0：类别索引（strand_break）
• x_center,y_center：中心点（归一化到 [0,1]）
• width,height：边界框宽高（归一化）

2. VOC 格式（.xml）

<object><name>strand_break</name><bndbox><xmin>450</xmin><ymin>200</ymin><xmax>570</xmax><ymax>260</ymax></bndbox></object>

✅ 可使用labelme或LabelImg导出为.xml，再用脚本转为 YOLO 格式。

✅ 三、训练配置文件：data.yaml

# data.yamltrain:./power_line_dataset/train/imagesval:./power_line_dataset/val/imagestest:./power_line_dataset/test/images# number of classesnc:1# class namesnames:['strand_break']

✅ 四、YOLOv8 完整训练代码：train_power_line.py

# train_power_line.pyimportosimportsysfromultralyticsimportYOLOimporttorchimportmatplotlib.pyplotaspltimportnumpyasnpdefmain():# 模型路径（可选预训练模型）model_path="yolov8n.pt"# 使用官方小模型# model_path = None # 从头训练# 数据配置data_yaml="data.yaml"# 训练参数epochs=200imgsz=640batch_size=16lr0=0.001weight_decay=0.0005workers=4device="cuda"iftorch.cuda.is_available()else"cpu"print(f"🚀 开始训练：{model_path}")print(f"⚙️ 设备：{device}")# 加载模型model=YOLO(model_path)ifmodel_pathelseYOLO("yolov8n.pt")# 开始训练results=model.train(data=data_yaml,epochs=epochs,imgsz=imgsz,batch=batch_size,lr0=lr0,wd=weight_decay,workers=workers,device=device,name="power_line_strand_break",project="power_line_detection",patience=10,# 早停save_period=10,# 每10轮保存一次augment=True,# 数据增强mosaic=0.5,# Mosaic 增强比例mixup=0.2,# MixUp 增强比例copy_paste=0.2,# Copy-Paste 增强degrees=15,# 旋转角度translate=0.1,# 平移scale=0.1,# 缩放shear=0.1,# 剪切flipud=0.2,# 上下翻转fliplr=0.5,# 左右翻转brightness=0.3,# 亮度调整contrast=0.3,# 对比度调整)# 验证模型print("📊 正在验证模型...")metrics=model.val(data=data_yaml,device=device)# 显示最终性能print(f"✅ 最终 mAP@0.5:{metrics.box.map50:.4f}")print(f"✅ 最终 mAP@0.5-0.95:{metrics.box.map:.4f}")# 绘制训练曲线plot_training_curves(results)defplot_training_curves(results):"""绘制训练曲线"""fig,axs=plt.subplots(2,2,figsize=(12,10))fig.suptitle("YOLOv8 Training Curves",fontsize=16)# 精度axs[0,0].plot(results.metrics.precision,label='Precision')axs[0,0].set_title('Precision')axs[0,0].legend()# 召回率axs[0,1].plot(results.metrics.recall,label='Recall')axs[0,1].set_title('Recall')axs[0,1].legend()# mAP@0.5axs[1,0].plot(results.metrics.mAP50,label='mAP@0.5')axs[1,0].set_title('mAP@0.5')axs[1,0].legend()# mAP@0.5-0.95axs[1,1].plot(results.metrics.mAP50_95,label='mAP@0.5-0.95')axs[1,1].set_title('mAP@0.5-0.95')axs[1,1].legend()plt.tight_layout()plt.savefig("training_curves.png",dpi=100)plt.show()if__name__=="__main__":main()

✅ 五、运行命令

python train_power_line.py

⚠️ 确保已安装依赖：

pipinstallultralytics matplotlib numpy torch torchvision

✅ 六、训练结果示例

1. 输出目录结构

runs/detect/power_line_strand_break/ ├── weights/ │ ├── best.pt # 最佳模型 │ └── last.pt # 最后一轮模型 ├── results.png # 训练曲线图 ├── train_batch.jpg # 训练样本 └── val_batch.jpg # 验证样本

2. 性能指标（预期）

✅ 如图所示，训练过程稳定收敛，达到高精度。

✅ 七、推理代码（可选）

# inference.pyfromultralyticsimportYOLO model=YOLO("runs/detect/power_line_strand_break/best.pt")results=model("path/to/test_image.jpg",conf=0.5,iou=0.5)results[0].show()# 显示带框图像