yolov5跑模型时候报 Class labels must be a dictionary of numbers to string解决方案

问题描述：在使用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']

至此，问题解决！！！