tfivea 2022-04-11 22:50 采纳率: 100%
浏览 1934
已结题

如何在训练过的yolov5模型中增加新的识别类

我最近在学习使用yolov5。
我用找到的数据集训练了识别头和头盔以后，怎么样才能在当前基础上新增加一个摩托车的类一起识别呢？

  • 写回答

1条回答 默认 最新

  • ty94666 2022-04-12 07:41
    关注

    你改过model的配置文件没，coco128.yaml和你用的模型（假设你用的yolov5s）yolov5s.yaml

    看你描述你应该是改过了，看看你模型目前能分多少类，如果只有2类，抱歉，只能全部重新训练。如果还有没用的类别，可以再训练时加载上次的best.pt就行

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

报告相同问题？

问题事件

  • 系统已结题 4月20日
  • 已采纳回答 4月12日
  • 创建了问题 4月11日