这是一个困扰我一周的bug,断断续续找了一周的问题,今天总算让我找出来问题所在了!
首先,如果你的yolov5在训练的时候出现这种情况:
labels = 0;
只有obj loss,cls loss和reg loss皆为0?
那么说明肯定出现了我所说的这个bug,建议可以试一试我的办法,先说解决办法:将cache文件删除,重新开始训练会自动新生城cache,重新建立索引表,即可找到所有的label;
那么为什么呢?
我的yolo训练方法是将images和labels单独放到两个文件夹里,labels是.txt格式,标准的训练yolo格式。然后生成train.txt,val.txt文件用来索引所有的图像文件,例如:
./images/val/000000239274.jpg
首先我是在公司共用的服务器上保存的coco数据集,所以会出现大家一起用的情况,可能这个cache里保存的就是上一个人用过,有过索引表的cache,跟你生成的train.txt,val.txt文件中的索引是不一样的,那么索引都对应错了,肯定label找不到了,如果没有label的话又怎么会有regress loss 和cls loss 呢?