使用YOLOv5训练自己的数据集时,遇到的问题

用于记录自己的一些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,行数不一定一样,可以用搜索找到对应位置,它这里默认的路径名是imageslabels,结合我之前的文件结构,可以发现我用的是Images而不是images,一字之差,导致错误。

解决方法

很简单,如果暂时只用yolov5进行一次训练的话,那大可直接把上面图的sa和sb改为对应的绝对路径
但是为了避免后面自己忘了这回事,还是老老实实地把文件结构修改好吧。
在这里插入图片描述改成这样以后就可以正常训练了。

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值