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

This is an official implementation for "Self-Supervised Learning with Swin Transformers".

Crossover Learning for Fast Online Video Instance Segmentation (ICCV 2021)

PySLM Python Library for Selective Laser Melting and Additive Manufacturing

KakaoBrain KoGPT (Korean Generative Pre-trained Transformer)

An implementation of "Learning human behaviors from motion capture by adversarial imitation"

This repository contains the code and models necessary to replicate the results of paper: How to Robustify Black-Box ML Models? A Zeroth-Order Optimization Perspective

Deep Inertial Prediction (DIPr)

DANA paper supplementary materials

CVPR 2021 - Official code repository for the paper: On Self-Contact and Human Pose.

A simple rest api serving a deep learning model that classifies human gender based on their faces. (vgg16 transfare learning)

Progressive Growing of GANs for Improved Quality, Stability, and Variation

Semi-supervised semantic segmentation needs strong, varied perturbations

This initial strategy was developed specifically for larger pools and is based on taking a moving average and deriving Bollinger Bands to create a projected active liquidity range.

🐸STT integration examples

Dynamics-aware Adversarial Attack of 3D Sparse Convolution Network

[NeurIPS 2020] Code for the paper "Balanced Meta-Softmax for Long-Tailed Visual Recognition"

This repo is about to create the Streamlit application for given ML model.

This repo is about implementing different approaches of pose estimation and also is a sub-task of the smart hospital bed project :smile:

Code and data (Incidents Dataset) for ECCV 2020 Paper "Detecting natural disasters, damage, and incidents in the wild".

Deeplab-resnet-101 in Pytorch with Jaccard loss