问题描述:
使用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】,问题解决!