如何使用 YOLOv8 训练防震锤高压塔线路行业数据集的详细步骤。这个数据集主要用于检测高压塔线路上的防震锤是否存在缺陷,数据集包含1000张图片,共2700+个样本,标签为“缺陷”和“未缺陷”,样本比为933:1808。
一、数据准备
- 数据集结构
假设你的数据集已经按照 VOC 格式组织,并且结构如下:
深色版本
power_line_dataset/
├── images/
│ ├── train/
│ │ ├── 000001.jpg
│ │ ├── 000002.jpg
│ │ └── …
│ ├── val/
│ │ ├── 000001.jpg
│ │ ├── 000002.jpg
│ │ └── …
│ └── test/
│ ├── 000001.jpg
│ ├── 000002.jpg
│ └── …
├── annotations/
│ ├── train/
│ │ ├── 000001.xml
│ │ ├── 000002.xml
│ │ └── …
│ ├── val/
│ │ ├── 000001.xml
│ │ ├── 000002.xml
│ │ └── …
│ └── test/
│ ├── 000001.xml
│ ├── 000002.xml
│ └── …
└── power_line.yaml
images/ 目录包含输入图像。
annotations/ 目录包含对应的标注文件,每个标注文件为 XML 格式。
power_line.yaml 是数据集的配置文件。
2. 配置文件
创建一个 power_line.yaml 文件,内容如下:
yaml
深色版本