最近，由于实验要求，我需要把在服务器上训练好的pytorch模型预训练.pth文件部署到安卓端测试推理时间，但是一直不知道应该怎么转变模型和部署，查了很多资料，遇到了很多问题，在同学的帮助下，尝试成功。
我简单记录一下整个部署流程，希望可以帮助想要尝试的同志们，同时，如果之后还需要部署相同项目的时候，我还可以参考这篇笔记。
一共有四个部分，pytorch->onnx->nn->Android
1.pytorch->onnx
（1）安装onnx和onnxruntime
onnx: onnxruntime:
CPU版本： GPU版本： 我使用的是gpu版本，在服务器上安装的
注意：在这个函数之前一定先使用model.eval（)
2)onnxruntime加载onnx模型测试
3）克隆nn项目
然后，编译一下
在build/tools/onnx/ 目录下就有转换工具onnx2nn文件了
然后去到目录build/tools/onnx/下运行（每次运行新的程序都需要重新执行上面的代码编译onnx）
mobilenetv3.param 和mobilenetv3.bin是我们所需的nn文件
这个链接里面也有教程，应该怎么使用，很好用。
然后sync project 或者重启工程
今天的文章 2026年onnx模型部署 python（onnx模型部署到手机）分享到此就结束了，感谢您的阅读。
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
如需转载请保留出处：https://bianchenghao.cn/bian-cheng-ri-ji/54602.html