无人机视角高速路面损害检测数据集与 YOLOv8 训练实战

注：总标注实例数为 7,640。单张图像可能包含多个类别。

数据划分

该数据集适用于无人机巡检系统、多类别路面病害联合检测及小目标识别任务。

环境准备

推荐使用 Python 3.8+ 环境，安装核心依赖库：

pip install ultralytics opencv-python numpy pandas tqdm

ultralytics 是官方 YOLOv8 库，原生支持 YOLO 格式，并可通过转换脚本兼容 VOC/COCO。

数据组织结构

建议按以下目录结构整理数据，确保每张图像 xxx.jpg 对应一个 xxx.txt 标签文件：

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

YOLO 格式标签内容示例（归一化坐标 [0,1]）：

<class_id> <x_center> <y_center> <width> <height>

若仅有 VOC 或 COCO 标注，可参考后文提供的转换脚本。

模型配置

创建 highway_damage.yaml 配置文件，定义路径与类别映射：

# highway_damage.yaml
train: ./HighwayDamageDataset/images/train
val: ./HighwayDamageDataset/images/val
test: ./HighwayDamageDataset/images/test
nc: 6
names:
  - Cracks
  - Waterlogging
  - Ravelling
  - Muddy_road
  - Road_side_garbage
  - Potholes

提示：虽然保留了英文命名，但推理时可映射为中文显示，避免部分工具兼容性问题。

训练脚本

以下是基于 ultralytics 的完整训练脚本，针对航拍图像特点优化了参数：

# train_highway.py
from ultralytics import YOLO
import os

def main():
    # 创建输出目录
    os.makedirs("runs/highway_damage", exist_ok=True)
    
    # 加载预训练模型（推荐 yolov8s 或 yolov8m）
    model = YOLO('yolov8s.pt')
    
    # 开始训练
    results = model.train(
        data='highway_damage.yaml',
        epochs=150,
        imgsz=1024,
        batch=8,
        name='yolov8s_highway_1024',
        project='runs/highway_damage',
        device=0,
        workers=8,
        cache=False,
        optimizer='AdamW',
        lr0=0.01,
        lrf=0.01,
        momentum=0.937,
        weight_decay=0.0005,
        warmup_epochs=3,
        patience=30,
        save=True,
        save_period=10,
        verbose=True,
        plots=True,
        hsv_h=0.015,
        hsv_s=0.7,
        hsv_v=0.4,
        degrees=10.0,
        translate=0.1,
        scale=0.5,
        shear=2.0,
        perspective=0.001,
        flipud=0.0,
        fliplr=0.5,
        mosaic=1.0,
        mixup=0.1
    )
    print(f"✅ 训练完成！最佳模型路径：{results.save_dir}/weights/best.pt")

if __name__ == '__main__':
    main()

也可使用命令行快速启动：

yolo detect train \
  data=highway_damage.yaml \
  model=yolov8s.pt \
  epochs=150 \
  imgsz=1024 \
  batch=8 \
  name=yolov8s_highway_1024 \
  project=runs/highway_damage \
  device=0 \
  optimizer=AdamW \
  patience=30 \
  hsv_h=0.015 hsv_s=0.7 hsv_v=0.4 \
  degrees=10 fliplr=0.5 mosaic=1.0

关键训练建议

格式转换脚本

VOC 转 YOLO

# voc2yolo.py
import os
import xml.etree.ElementTree as ET
from pathlib import Path

class_names = [
    "Cracks", "Waterlogging", "Ravelling", 
    "Muddy_road", "Road_side_garbage", "Potholes"
]

def convert_voc_to_yolo(voc_dir, yolo_dir, image_dir):
    os.makedirs(yolo_dir, exist_ok=True)
    for xml_file in Path(voc_dir).glob("*.xml"):
        tree = ET.parse(xml_file)
        root = tree.getroot()
        img_w = int(root.find('size/width').text)
        img_h = int(root.find('size/height').text)
        lines = []
        for obj in root.findall('object'):
            cls_name = obj.find('name').text
            if cls_name not in class_names:
                continue
            cls_id = class_names.index(cls_name)
            bndbox = obj.find('bndbox')
            xmin = int(bndbox.find('xmin').text)
            ymin = int(bndbox.find('ymin').text)
            xmax = int(bndbox.find('xmax').text)
            ymax = int(bndbox.find('ymax').text)
            x_center = (xmin + xmax) / 2 / img_w
            y_center = (ymin + ymax) / 2 / img_h
            width = (xmax - xmin) / img_w
            height = (ymax - ymin) / img_h
            # 注意：YOLO 格式需空格分隔
            lines.append(f"{cls_id} {x_center:.6f} {y_center:.6f} {width:.6f} {height:.6f}")
        with open(os.path.join(yolo_dir, xml_file.stem + '.txt'), 'w') as f:
            f.write('\n'.join(lines))

# 使用示例
# convert_voc_to_yolo('annotations/voc/train', 'labels/train', 'images/train')

COCO 转 YOLO

YOLOv8 原生支持 COCO，只需在 highway_damage.yaml 中指定 JSON 路径，并确保类别顺序一致。

推理与部署

from ultralytics import YOLO

model = YOLO('runs/highway_damage/yolov8s_highway_1024/weights/best.pt')
results = model('test_image.jpg')

chinese_names = {
    "Cracks": "裂缝",
    "Waterlogging": "积水",
    "Ravelling": "松散",
    "Muddy_road": "泥泞道路",
    "Road_side_garbage": "道路旁垃圾",
    "Potholes": "坑洼"
}

for r in results:
    boxes = r.boxes
    for box in boxes:
        cls_name = model.names[int(box.cls)]
        print(f"检测到：{chinese_names.get(cls_name, cls_name)}")

注意：由于图像宽高比极端（1152×2048 ≈ 9:16），YOLO 会自动 padding 成正方形。若需保留原始比例，可考虑滑动窗口裁剪或使用 YOLOv8-OBB（本任务无需旋转框）。