用yolov5训练自己的数据集出现的问题

整个修改过程参考:yolov5训练自己的数据集
1.ValueError:expected sequence of length 6 at dim but got 5(类似这样的问题)

问题分析:当自己修改model中yaml文件中聚类锚框ratio的时候,不知道什么原因,修改成自己生成的就会出现上述错误。
如果有问题,可以先用默认的数据就可以了。
在这里插入图片描述
2.YoloV5训练本地数据问题解决:AssertionError: train: No labels in 2007_train.cache. Can not train without labels

上述问题是由自己的放置数据集的文件夹和dataset中的数据集名称不同。
在dataset.py中(ctrl+F)搜索define label
在这里插入图片描述
将上图中文件夹的名字改成自己设置的,如果不一致就会出现以上问题。


版权声明:本文为further_Yan原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。