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

[SDM 2022] Towards Similarity-Aware Time-Series Classification

Implements Gradient Centralization and allows it to use as a Python package in TensorFlow

A clean and robust Pytorch implementation of PPO on continuous action space.

Code for Motion Representations for Articulated Animation paper

A simple, clean TensorFlow implementation of Generative Adversarial Networks with a focus on modeling illustrations.

Multiband spectro-radiometric satellite image analysis with K-means cluster algorithm

Fibonacci Method Gradient Descent

An implementation for the loss function proposed in Decoupled Contrastive Loss paper.

Neural Nano-Optics for High-quality Thin Lens Imaging

Some experiments with tennis player aging curves using Hilbert space GPs in PyMC. Only experimental for now.

Real time Human Detection Counting

Pytorch implementation for DFN: Distributed Feedback Network for Single-Image Deraining.

Code to compute permutation and drop-column importances in Python scikit-learn models

[ECCV 2020] XingGAN for Person Image Generation

Semantic-aware Grad-GAN for Virtual-to-Real Urban Scene Adaption

Neural-fractal - Create Fractals Using Complex-Valued Neural Networks!

Learned model to estimate number of distinct values (NDV) of a population using a small sample.

Implementation of the paper titled "Using Sampling to Estimate and Improve Performance of Automated Scoring Systems with Guarantees"

A framework for using LSTMs to detect anomalies in multivariate time series data. Includes spacecraft anomaly data and experiments from the Mars Science Laboratory and SMAP missions.

Open source repository for the code accompanying the paper 'Non-Rigid Neural Radiance Fields Reconstruction and Novel View Synthesis of a Deforming Scene from Monocular Video'.