YOLO 提供默认的标注工具:
pip install labelimg
安装后,执行 labelimg
打开标注工具。
数据集和标签选择:
自定义数据集再 datasets 下创建指定的数据集和名称。
.
|____datasets
| |____coco128
| | |____LICENSE
| | |____images
| | | |____train2017
| | | | |____000000000009.jpg
| | | | |____000000000025.jpg
| | |____labels
| | | |____train2017.cache
| | | |____train2017
| | | | |____000000000009.txt
| | | | |____000000000025.txt
再 datasets 下创建 coco128.yaml
文件,内容如下:
path: /path/coco128
train: images/train2017
val: images/train2017
nc: 80
names: [person, bicycle, car, motorcycle, airplane, bus, ...]
或
names:
0: 'person'
1: 'bicycle'
2: 'car'
3: 'motorcycle'
4: 'airplane'
5: 'bus'
nc: 6
train: /path/train
val: /path/val