qq_66048450 2023-04-23 21:55 采纳率: 33.3%
浏览 91
已结题

yolov5训练模型出问题

YOLOv5-6.1训练一个四分类模型时出现如下错误,请问大家如何修改

  File "F:\yolo\road_detect\yolov5-v6.1\utils\loss.py", line 148, in __call__
   t[range(n), tcls[i]] = self.cp
IndexError: index 9 is out of bounds for dimension 1 with size 4

img

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2023-04-24 09:22
    关注

    分类请运行yolov5/classify/train.py这个文件,而不是yolov5/train.py,这两个是不一样的

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
    1人已打赏
查看更多回答(1条)

报告相同问题?

问题事件

  • 系统已结题 5月2日
  • 已采纳回答 4月24日
  • 创建了问题 4月23日