yolov5中触发StopIteration问题

问题遇到的现象和发生背景

用gpu跑yolov5训练自己的图像,无法训练,从train.py文件中 plot_labels(labels, names, save_dir)返回StopIteration

img

结果:

img

我的解答思路和尝试过的方法

曾经调整过parse的参数

我想要达到的结果

希望可以正常运行

看下你的matplotlib版本多少,如果是3.6.x,降级到3.5看下
pip install matplotlib==3.5.0