训练自定义数据
- 新建dataset.yaml
COCO128是一个小型教程数据集,由COCOtrain2017中的前128张图像组成。这128张图片用于训练和验证,以验证我们的训练通道是否能够拟合。数据集配置文件如下所示:它定义了
1)一个可选的下载命令/URL,用于自动下载。
2)训练图像目录的路径(或带有训练图像列表的txt文件路径)
3)用于验证的图像
4)类的数量
5)类名列表
download: https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip
train: ../coco128/images/train2017/
val: ../coco128/images/train2017/
nc: 80
names: ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light',
'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow',
'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee',
'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard',
'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch',
'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard',
'cell phone', 'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors',
'teddy bear', 'hair drier', 'toothbrush']
- 创建标签
使用labelbox之类的工具标记图像后,将标签导出为YOLO
格式,每个图像带有一个txt文件(如果图像中没有对象,则不需要txt文件)。