yolov5模型通过Tengine框架部署在FT2000上全记录
注意:如果只用原参数转为.onnx文件,会出现一些Tengine不支持的算子,因此我们需要用onnx的内置的一些api,在我们转换时,要加上 --train=True 和 --simplifier,加上后我们转换后的onnx文件才不会包含一些Tengine框架不支持的算子,并且取消掉了后处理操作。分为几步:不管用什么框架训练的网络模型都可以首先把我们训练好的权重文件,比如我用的pytorch框架做的训练,那我就将.pt文件转换为.onnx文件。注意:转换工具只能在x86的环境下编译,才能生成。
原创
2023-04-03 10:25:25 ·
926 阅读 ·
2 评论