使用yolov5-7.0目标检测报错

Question

windows，环境配置好了，detect.py运行正常，但是默认使用cpu，需要加--device 0才能用gpu跑。主要问题是train.py报错不会改，如图求大佬帮忙

Answer 1

引自免费微信小程序：皆我百晓生

从错误信息来看，问题出在data.yaml文件中的names部分。错误提示表明程序期望data['names']下的键应该是整数类型，但实际上遇到了字符串类型，导致尝试访问.key()方法时出现了AttributeError，因为字符串对象没有.key()这个属性。

解决这个问题，你需要确保data.yaml文件中的names部分是按照正确的格式编写的。正确的格式应该是每个类别名称前面都有一个整数索引，形如：

nc: 80  # number of classes
names:
  0: person
  1: bicycle
  2: car
  ...

这里的0, 1, 2等就是类别ID，它们应当是整数，并且紧跟着冒号和类别名称。

请检查并修改你的data.yaml文件，确保所有的类别名称前都有正确的整数索引，并且这些索引确实是整数类型，没有被误写为字符串或其他格式。如果你的文件中类别名直接以字符串列表形式列出（例如 - car），那么需要调整为上面所示的格式。

修改完成后，再次尝试运行train.py，应该可以顺利进行训练了。如果还有其他问题，不妨再提供更详细的错误信息，我会尽力帮助你解决。