Zen的小站

小舟从此逝,江海寄余生

0%

【深度学习工具】onnx学习

onnx的pytorch模型部署方法

torch模型转onnx

model.eval()output = modle(input)后面添加一句话:

torch.onnx.export(model, input, 'modelname.onnx')即可导出

onnx模型的使用

​ 不知。。

​ 有空看看部署在安卓设备的方法。(我们新生项目小组已经做好了,用的是maui+onnx )

(105条消息) 在Android设备部署PyTorch模型_紫芝的博客-CSDN博客_android pytorch

(105条消息) 将Pytorch模型部署到Android端_YDDUONG的博客-CSDN博客_pytorch模型部署到安卓

maui

electron

onnx模型的优点

不知。。

多想多做,发篇一作

-------------本文结束感谢您的阅读-------------
// 在最后添加