我做的是桌椅分类
在创建完虚拟环境配完包时,如果能跑通代码就去下载数据集并用标注工具标注(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都行
改为自己的数据集前一定能跑通原来的代码
标签:yolov5,train,py,yaml,改为,数据,best,标注 From: https://blog.csdn.net/2301_79571396/article/details/139429015