用于记录自己的一些bug经历
如果需要在自己的数据集上面训练YOLOv5,那么首先要参考官方教程YOLOv5官方教程
一、有关创建train.cache的报错
因为在写本文时,电脑正在用于训练模型,无法展示报错情况,后面再添加上去。
我的问题产生的原因很简单,文件目录名的设置不对,所以一定要先看一遍官方的教程!!
我原本的文件目录如下
这个文件结构参考了文章
【YOLOV5】记录一次自己在yolo训练模型上的全过程(附上多次解决Bug记录以及心得体会)
感谢博主给我提供的思路!!
稍微解释一下各个文件夹的内容
Annotations:标注数据的VOC格式,里面是对应的xml文件
dataSet_path:包含train.txt,test.txt,val.txt三个文件,记录着
数据集图片的绝对路径
Images:数据集所有的图片
ImageSets:记录着数据集所有图片的文件名(无后缀,且不包含路径)
labels:标注数据的txt格式,里面存储着txt文件
这个时候如果直接运行train.py文件,那么会报错,报错内容是关于cache文件的创建。如果文件结构和我之前一样的,那么大概率问题出自文件名不规范。
报错图
后面训练结束了我再加上去。
错误出现原因:
如上图,这个部分属于yolov5-master下的
util/dataloaders.py
,行数不一定一样,可以用搜索找到对应位置,它这里默认的路径名是images
和labels
,结合我之前的文件结构,可以发现我用的是Images而不是images,一字之差,导致错误。
解决方法
很简单,如果暂时只用yolov5进行一次训练的话,那大可直接把上面图的sa和sb改为对应的绝对路径。
但是为了避免后面自己忘了这回事,还是老老实实地把文件结构修改好吧。
改成这样以后就可以正常训练了。