AssertionError: Label class 1 exceeds nc=1 in data/steel.yaml. Possible class labels are 0-0

在使用自己的数据集运行yolov5程序时报错:
AssertionError: Label class 1 exceeds nc=1 in data/steel.yaml. Possible class labels are 0-0

AssertionError: Label class 1 exceeds nc=1 in data/steel.yaml. Possible class labels are 0-0_第1张图片
仔细阅读报错内容可以发现:类别与标签序号不匹配(博主的自定义数据集只有一类)
经过博主的仔细思考,猜测是在生成labels时出现了问题,通过阅读生成labels的脚本,找到问题如下:AssertionError: Label class 1 exceeds nc=1 in data/steel.yaml. Possible class labels are 0-0_第2张图片
解决方法:将classes列表中的person字符串删除即可!!!

你可能感兴趣的:(深度学习项目运行报错,深度学习,目标检测,计算机视觉,pytorch)