已解决训练自己yolov7检测模型时报错：AssertionError: train: No labels in XXX/XXX/train.cache.
问题描述
在使用yolov7训练自己数据集的模型时，数据集制作与处理完成以及对train.py文件的参数也修改完成之后，运行train.py出现报错内容为：AssertionError: train: No labels in VOCdevkit/VOC2007/train.cache.（VOCdevkit/VOC2007是我存放自己数据集的路径）
解决方法
-
进行调试后，发现yolov7源代码在utils/datasets.py文件中的def img2label_paths(img_paths): 函数中默认存放数据集图像的文件夹是images，需要将’images’为’JPEGImages’（自己存放数据集图片的文件夹），就能根据图像位置获得标签位置。
-
但是经过上步修改后运行train.py还会报相同的错误时，这是因为前面运行了train.py文件，已在VOCdevkit/VOC2007文件夹下生成train.cache文件（不完整的），需要在下次运行前将train.cache文件删除。
-
经过上面两个操作后重新运行train.py网络就开始训练了。
文章出处登录后可见！
已经登录？立即刷新
