Ucla_MIT
2021-10-19 14:47
采纳率: 0%
浏览 89
已结题

yolo预训练模型导出为onnx格式报错

yolov5添加了psp分割头,使用export.py转化预训练模型为onnx格式报错,具体报错如图

img

img

img

  • 好问题 提建议
  • 收藏

1条回答 默认 最新

  • ProfSnail 2021-10-20 23:45

    你是不是把预训练模型的参数放到一个list里面了?
    建议检查代码内容,把list里面的内容取出来,合并转化为tensor格式。

    评论
    解决 无用
    打赏 举报

相关推荐 更多相似问题