问题描述:在使用yolov5跑目标检测模型时,报Class labels must be a dictionary of numbers to string是由于类标签不匹配的问题。
解决方法:
1.找到data目录下的helmet.yaml中的names,如下所示:
代码如下:
names:
0:helmet
1: nohelmet
2.将步骤1中的names修改为如下图所示:
代码如下:
names:['helmet','nohelmet']
至此,问题解决!!!