用gpu跑yolov5训练自己的图像,无法训练,从train.py文件中 plot_labels(labels, names, save_dir)返回StopIteration
结果:
曾经调整过parse的参数
希望可以正常运行
看下你的matplotlib版本多少,如果是3.6.x,降级到3.5看下pip install matplotlib==3.5.0