yolov5改为自己的数据集

我做的是桌椅分类

在创建完虚拟环境配完包时，如果能跑通代码就去下载数据集并用标注工具标注(labelimg)我选择的格式是.txt格式标注晚并保存。

yolov5中是我标注好的数据集其中images/train2017是标注好的图片labels/train2017是存放的是标注好的.txt格式的图片

在data中images的图片改为自己标注好的

在data中创建mydata.yaml复制data中的coco128.yaml并修改path、train、val

在models中创建mydata.yaml复制使用的yaml如yolov5l.yaml、yolov5m.yaml、yolov5n.yaml、yolov5s.yaml、yolov5x.yaml等修改分类个数.我这里使用的是yolov5s.yaml所以复制它。

修改train.py

这个地方改为自己的“自己的.yaml”文件

修改val.py

这个地方改为自己的“自己的.yaml”文件

best.pt是在train.py训练完后出现的

将best.py拖到yolo5-master下重构就行

修改detect.py

这个地方改为自己的“自己的.yaml”文件

先运行train.py运行完重构best.py在运行detect.py或test.py都行

改为自己的数据集前一定能跑通原来的代码