标签：转换 onnx torch simplify pytorch input model

torch与onnx转换

一、pytorch转onnx

pytorch官方已经提供了支持。

import torch

dummy_input = torch.randn(1, 3, 480, 640)
torch.onnx.export(model, dummy_input, 'net_640x480.onnx', export_params=True, verbose=False, training=False, input_names=['Input_1'], output_names=['Concat_211', 'Reshape_235'], opset_version=11, dynamic_axes={"Input_1":{0:"batch_size"}, "Concat_211":{0:"batch_size"}, "Reshape_235":{0:"batch_size"}})

导出onnx后，可以使用onnx-simplify去掉模型的一些冗余操作：

from onnxsim import simplify
import onnx

onnx_model = onnx.load("net_640x480.onnx")
model_simp, check = simplify(onnx_model, input_shapes={'Input_1':[1, 3, 480, 640]})
onnx.save(model_simp, 'net_640x480_simplify.onnx')

二、onnx转pytorch

可以使用第三方工具onnx2torch:

https://github.com/fumihwh/onnx-pytorch.git

查看代码：

也可以参考：
https://github.com/ToriML/onnx2pytorch/tree/master/onnx2pytorch

三、其它

可以使用在线转换工具进行模型的转换：

标签：转换,onnx,torch,simplify,pytorch,input,model
来源： https://www.cnblogs.com/gr-nick/p/15774334.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。