问题描述:

使用YOLOv5在训练自己的数据集时,运行【trian.py】文件时,出现了下面的报错:

主要提示为:

AssertionError: train: No labels in /data_1T/xd1/yolov5-6.1/datasets/VOC_To_YOLO/train.cache. Can not train without labels. See https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data


问题解决

这是因为【datasets.py】中的图片路径设置的不对

解决方案如下:

1.打开【datasets.py】。【datasets.py】一般在【utils】文件下。如图:

2.打开之后,按【Ctrl】+【F】打开搜索功能,输入【Define label 】找到下面的一行内容: 3.去看你图片的图片路径。例如,我的训练图片放在了这个位置:

 路径为:xx/xx/yolov5-6.1/datasets/VOC_To_YOLO/JPEGImages

4.修改【datasets.py】。需要把【JPEGImages】替换掉【datasets.py】里面的这个位置:

 最后的效果为:

5.保存刚才的修改操作 

6.再次运行【trian.py】,问题解决!


到这,就已经解决了问题。欢迎关注蓝胖胖,咱们一起解决BUG~~~

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐