yolov5 5.0训练时发生错误,但是6.1却可以训练
错误如下: names = ['item'] if opt.single_cls and len(data_dict['names']) != 1 else data_dict['names'] # class names
KeyError: 'names'
我只训练了一个目标
已经解决了,是yolo没找到训练的yaml文件改下路径就好了
data_dict['names']这里出错。
data_dict中有key为‘names’吗?
data_dict['names']应该去掉引号,改成 data_dict[names]