Convert Pytorch model to onnx or tflite, and the converted model can be visualized by Netron

Last update: Nov 19, 2022

Related tags

Deep Learning model-converter

Overview

Model converter

Convert Pytorch model to onnx or tflite, and the converted model can be visualized by Netron

You can use this project to:

Pytorch -> onnx (float32)
Pytorch -> onnx -> tflite (float32)
Pytorch -> onnx -> tflite (int8)

Requirements

torch2onnx

pytorch
onnx
opencv-python

torch2tflite

tensorflow ~= 2.5
torch == 1.8.1
tensorflow-addons ~= 0.15
opencv-python ~= 4.5.4
onnx ~= 1.10
onnx-tf ~= 1.9
numpy >= 1.19

(opencv-python is optional)

Usage

torch2onnx (float32)

from converter import Torch2onnxConverter

converter = Torch2onnxConverter(model_path, target_shape=(3,224,224))
converter.convert()

torch2tflite (float32)

from converter import Torch2TFLiteConverter

converter = Torch2TFLiteConverter(tmp_path, tflite_model_save_path='model_float32.lite', target_shape=(224,224,3))
converter.convert()

torch2tflite (int8)

from converter import Torch2TFLiteConverter

converter = Torch2TFLiteConverter(tmp_path, tflite_model_save_path='model_int8.lite', target_shape=(224,224,3),
                                    representative_dataset=representative_dataset)
converter.convert()

More details can be found in Torch2onnxConverter and Torch2TfliteConverter __init__ method.

Note that target_shape is different for Pytorch and Tensorflow.

Convert Pytorch model to onnx or tflite, and the converted model can be visualized by Netron

Related tags

Overview

Model converter

Requirements

Usage

Example

Owner

Roxbili

Random-Afg - Afghanistan Random Old Idz Cloner Tools

Source Code and data for my paper titled Linguistic Knowledge in Data Augmentation for Natural Language Processing: An Example on Chinese Question Matching

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.

⚾🤖⚾ Automatic baseball pitching overlay in realtime

YOLTv5 rapidly detects objects in arbitrarily large aerial or satellite images that far exceed the ~600×600 pixel size typically ingested by deep learning object detection frameworks

From Fidelity to Perceptual Quality: A Semi-Supervised Approach for Low-Light Image Enhancement (CVPR'2020)

Anomaly detection analysis and labeling tool, specifically for multiple time series (one time series per category)

Codebase of deep learning models for inferring stability of mRNA molecules

Codecov coverage standard for Python

Computationally efficient algorithm that identifies boundary points of a point cloud.

Automatic voice-synthetised summaries of latest research papers on arXiv

CSE-519---Project - Job Title Analysis (Project for CSE 519 - Data Science Fundamentals)

mlpack: a scalable C++ machine learning library --

YOLOPのPythonでのONNX推論サンプル

[ICLR 2022] Pretraining Text Encoders with Adversarial Mixture of Training Signal Generators

Utilities and information for the signals.numer.ai tournament

Effect of Deep Transfer and Multi task Learning on Sperm Abnormality Detection

The official project of SimSwap (ACM MM 2020)

OCR Streamlit App is used to extract text from images using python's easyocr, pytorch and streamlit packages

Apply our monocular depth boosting to your own network!