欧亿体育
工作动态
我的位置: 首页 > 工作动态
一步真实解决AssertionError: train: No labels in /xxx/xxx/xxx/datasets/VOC_To_YOLO/train.cache.
发布时间:2024-01-14 06:02
  |  
阅读量:
  |  
作者:
欧亿体育

问题描述:

使用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~~~