一、大体的方式
1.标注
2.自己获得数据集(手动)-人工标注
3.自己获得数据集-半人工标注
4.仿真数据集(GAN,数字图像处理的方式)
二、人工制作数据集网站
Make Sensehttps://www.makesense.ai/三、步骤
Train Custom Data - YOLOv8 Docs (ultralytics.com)https://docs.ultralytics.com/yolov5/train_custom_data/#12-create-labels 参考上数据集制作步骤,上步骤更为详细,且运行yolov5代码也为上代码。
1.自行准备图片并制作标签
2.导入图片及标签,对car、person、bike...进行标注
3.导出数据集
4.将数据集和标签导入yolov5程序中进行检验
5.修改parser.add_argument('--data', type=str, default='data/coco128.yaml', help='data.yaml path')中data/coco128.yaml文件的参数,按照制作数据集时的标签进行修改。我复制出了一个mydata进行标签和测试训练数据集位置的修改。
6.进行train.py的训练,获得训练权重的数据集,如下所示。
7.选择训练好的权重模型best.pt,图片选择之前用的训练图片,修改detect.py的参数如图所示,进行训练即可完成。