YOLO标注以及自定义数据集.md 1.5 KB

YOLO标注工具

YOLO 提供默认的标注工具:

pip install labelimg

安装后,执行 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

参考