智慧巡检-管道检测数据集 YOLOV8模型如何训练-管道检测数据集
智慧巡检-管道检测数据集,3096张,
yolo和voc两种标注方式
图像尺寸:640*640
训练集图像数量:2160; 验证集图像数量:621; 测试集图像数量:315
类别名称: 每一类图像数 ,每一类标注数
pipe: 3015,6292
image num: 3096
管道检测数据集(YOLO+VOC双格式)+ YOLOv8 训练全套代码
我给你标准论文级表格+直接可运行训练代码+数据集结构,完全匹配你提供的 3096 张管道数据。
一、管道检测数据集 完整信息表(可直接放入论文/毕设)
1. 数据集基础信息
| 项目 | 内容 |
|---|---|
| 数据集名称 | 管道缺陷/管道目标检测数据集 |
| 总图像数量 | 3096 张 |
| 图像尺寸 | 640×640 标准尺寸 |
| 标注格式 | YOLO(txt) + VOC(xml) 双格式标注 |
| 任务类型 | 单类别目标检测(管道定位/管道异常检测) |
| 数据划分 | 训练集 2160 张 / 验证集 621 张 / 测试集 315 张 |
| 适用模型 | YOLOv8 / YOLOv11 / YOLOv26 / Faster R-CNN |
2. 类别与标注统计
| 类别ID | 类别名称 | 包含图像数量 | 标注框总数 |
|---|---|---|---|
| 0 | pipe(管道) | 3015 张 | 6292 个 |
| 汇总 | —— | 3096 张 | 6292 个 |
3. 数据集标准目录结构
pipe_dataset/ ├── images/ │ ├── train/ 2160张 │ ├── val/ 621张 │ └── test/ 315张 ├── labels/ │ ├── train/ YOLO txt标注 │ ├── val/ │ └── test/ ├── annotations/ VOC xml标注 └── pipe_data.yaml二、YOLO 数据集配置文件pipe_data.yaml
path:./pipe_datasettrain:images/trainval:images/valtest:images/testnc:1names:0:pipe三、YOLOv8 训练代码(直接运行)
train_pipe.py
fromultralyticsimportYOLOimporttorch# 管道目标检测训练if__name__=="__main__":# 加载YOLOv8模型(n/s/m/l 任选,推荐s)model=YOLO("yolov8s.pt")# 训练参数(工业管道场景优化)model.train(data="pipe_data.yaml",# 数据集配置epochs=120,# 训练轮数imgsz=640,# 图像尺寸 640*640batch=16,# 批次device=0iftorch.cuda.is_available()else"cpu",lr0=0.01,# 初始学习率patience=15,# 早停augment=True,# 数据增强mosaic=1.0,mixup=0.1,hsv_h=0.015,hsv_s=0.5,hsv_v=0.3,degrees=10,scale=0.3,name="pipe_detection",# 任务名称cache=True,amp=True)# 测试集评估model.val(split="test")四、快速运行命令
1. 安装依赖
pipinstallultralytics opencv-python torch torchvision2. 开始训练
python train_pipe.py3. 推理测试(训练完后用)
yolo predictmodel=./runs/detect/pipe_detection/weights/best.ptsource=test.jpgimgsz=640