Library to transform onnx model to pytorch.
Project description
ONNX to PyTorch
A library to transform ONNX model to PyTorch. This library enables use of PyTorch backend and all of its great features for manipulation of neural networks.
Installation
pip install onnx2pytorch
Usage
import onnx
from onnx2pytorch import ConvertModel
onnx_model = onnx.load(path_to_onnx_model)
pytorch_model = ConvertModel(onnx_model)
Currently supported and tested models from onnx_zoo:
- MobileNet
- ResNet
- ShuffleNet_V2
- BERT-Squad
- EfficientNet-Lite4
- Fast Neural Style Transfer
- Super Resolution
- YOLOv4 (Not exactly the same, nearest neighbour interpolation in pytorch differs)
- U-net (Converted from pytorch to onnx and then back)
Limitations
Known current version limitations are:
batch_size > 1
could deliver unexpected results due to ambiguity of onnx's BatchNorm layer.
That is why in this case for now we raise an assertion error.
Setexperimental=True
inConvertModel
to be able to usebatch_size > 1
.- Fine tuning and training of converted models was not tested yet, only inference.
Development
Dependency installation
pip install -r requirements.txt
From onnxruntime>=1.5.0 you need to add the
following to your .bashrc or .zshrc if you are running OSx:
export KMP_DUPLICATE_LIB_OK=True
Code formatting
The Uncompromising Code Formatter: Black
black {source_file_or_directory}
Install it into pre-commit hook to always commit nicely formatted code:
pre-commit install
Testing
Test fixtures
To test the complete conversion of an onnx model download pre-trained models:
./download_fixtures.sh
Use flag --all
to download more models.
Add any custom models to ./fixtures
folder to test their conversion.
Debugging
Set ConvertModel(..., debug=True)
to compare each converted
activation from pytorch with the activation from onnxruntime.
This helps identify where in the graph the activations start to differ.
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
Built Distribution
Hashes for onnx2pytorch-0.5.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ccb39f0e94a14115d52bd965e3c64f151dffd4aaa52f9cdace7067c8ebf2b9e3 |
|
MD5 | f30fc58ce950390905ecb1bc07555a2b |
|
BLAKE2b-256 | 74fb2c3e825ab81ac8fc0777f332ebab51f73c87fa3a6ba124981de8c0f13656 |