Convert onnx to pytorch code.
Project description
onnx-pytorch
Generating pytorch code from ONNX.
Currently support onnx==1.9.0
and torch==1.8.1
.
Usage
from onnx_pytorch import code_gen
code_gen.gen("/path/to/onnx_model", "/path/to/output_dir")
A model.py
file and variables
folder will be created under output_dir
.
Tutorial
- Download resnet18 onnx model
wget https://github.com/onnx/models/raw/master/vision/classification/resnet/model/resnet18-v2-7.onnx
- Use onnx-pytorch to generate pytorch code and variables.
from onnx_pytorch import code_gen
code_gen.gen("resnet18-v2-7.onnx", "./")
- Test result
import numpy as np
import onnx
import onnxruntime
import torch
torch.set_printoptions(8)
from model import Model
model = Model()
model.eval()
inp = np.random.randn(1, 3, 224, 224).astype(np.float32)
with torch.no_grad():
torch_outputs = model(torch.from_numpy(inp))
onnx_model = onnx.load("resnet18-v2-7.onnx")
sess_options = onnxruntime.SessionOptions()
session = onnxruntime.InferenceSession(onnx_model.SerializeToString(),
sess_options)
inputs = {"data": inp}
ort_outputs = session.run(None, inputs)
print(
"Comparison result:",
np.allclose(torch_outputs.detach().numpy(),
ort_outputs[0],
atol=1e-5,
rtol=1e-5))
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
onnx-pytorch-0.1.2.tar.gz
(33.6 kB
view hashes)
Built Distribution
Close
Hashes for onnx_pytorch-0.1.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d23040297d9a5bf18448d860b1465641da0d78aeb405d6a934b733d7d26c73c0 |
|
MD5 | 6c92da906bf7ba123e8459cc83db2c3a |
|
BLAKE2b-256 | 9a472bafe730583cbe4b81eaf1ad4b7ec8d45724d740035552accfb6d015420d |