问题遇到的现象和发生背景
用gpu跑yolov5训练自己的图像,无法训练,从train.py文件中 plot_labels(labels, names, save_dir)返回StopIteration
结果:
我的解答思路和尝试过的方法
曾经调整过parse的参数
我想要达到的结果
希望可以正常运行
用gpu跑yolov5训练自己的图像,无法训练,从train.py文件中 plot_labels(labels, names, save_dir)返回StopIteration
结果:
曾经调整过parse的参数
希望可以正常运行
看下你的matplotlib版本多少,如果是3.6.x,降级到3.5看下
pip install matplotlib==3.5.0