little shea 2022-10-17 01:25 采纳率: 20%
浏览 124
已结题

yolov5中触发StopIteration问题

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

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

img

结果：

img

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

曾经调整过parse的参数

我想要达到的结果

希望可以正常运行

  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2022-10-17 13:14
    关注

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

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题？

问题事件

  • 系统已结题 10月25日
  • 已采纳回答 10月17日
  • 创建了问题 10月17日