问题描述
在使用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文件删除。
文章来源:https://www.toymoban.com/news/detail-460242.html -
经过上面两个操作后重新运行train.py网络就开始训练了。文章来源地址https://www.toymoban.com/news/detail-460242.html
到了这里,关于已解决训练自己yolov7检测模型时报错:AssertionError: train: No labels in XXX/XXX/train.cache.的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!