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

Variational autoencoder for anime face reconstruction

Project ArXiv Citation Network

Data and code for the paper "Importance of Kernel Bandwidth in Quantum Machine Learning"

Genshin-assets - 👧 Public documentation & static assets for Genshin Impact data.

DeepHawkeye is a library to detect unusual patterns in images using features from pretrained neural networks

PrimitiveNet: Primitive Instance Segmentation with Local Primitive Embedding under Adversarial Metric (ICCV 2021)

Zero-Cost Proxies for Lightweight NAS

Hand-distance-measurement-game - Hand Distance Measurement Game

Space-invaders - Simple Game created using Python & PyGame, as my Beginner Python Project

Pytorch implementation of the paper "Enhancing Content Preservation in Text Style Transfer Using Reverse Attention and Conditional Layer Normalization"

Short and long time series classification using convolutional neural networks

An Implementation of Fully Convolutional Networks in Tensorflow.

DEMix Layers for Modular Language Modeling

Referring Video Object Segmentation

Pre-trained model, code, and materials from the paper "Impact of Adversarial Examples on Deep Learning Models for Biomedical Image Segmentation" (MICCAI 2019).

Tensors and neural networks in Haskell

Minimal But Practical Image Classifier Pipline Using Pytorch, Finetune on ResNet18, Got 99% Accuracy on Own Small Datasets.

Code for the paper "Spatio-temporal Self-Supervised Representation Learning for 3D Point Clouds" (ICCV 2021)

Python code to generate art with Generative Adversarial Network

An open source implementation of CLIP.