search

Wulingtian / yolov5_onnx2caffe

yolov5 onnx caffe
113 stars 37 forks source link

> 我也遇到了，解决办法在.pt文件转换onnx的时候加上--train； #14

Closed huangzongmou closed 2 years ago

huangzongmou commented 2 years ago

我也遇到了，解决办法在.pt文件转换onnx的时候加上--train； eg:"python ./models/export.py --weights ./models/.pt --img-size --batch 1 --opset-version 10 --train";可以这样尝试能不能解决KeyError: '395'的错误

感谢，通过加上--train，去掉了网络结构最后的后处理部分，转换成功了。

_Originally posted by @wllkk in https://github.com/Wulingtian/yolov5_onnx2caffe/issues/9#issuecomment-862171037_

heylary commented 2 years ago

请问我加--train提示没有这个参数，是我的版本问题吗？

huangzongmou commented 2 years ago

您好！您的邮件已收到，尽快给你回复。谢谢                                                      黄总谋！