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

BT-Unet: A-Self-supervised-learning-framework-for-biomedical-image-segmentation-using-Barlow-Twins

N-Person-Check-Checker-Splitter - A calculator app use to divide checks

Code for the ICASSP-2021 paper: Continuous Speech Separation with Conformer.

Automate issue discovery for your projects against Lightning nightly and releases.

Vanilla and Prototypical Networks with Random Weights for image classification on Omniglot and mini-ImageNet. Made with Python3.

This repository contains implementations and illustrative code to accompany DeepMind publications

Continuous Conditional Random Field Convolution for Point Cloud Segmentation

This is my research project for the Irving Center for Cancer Dynamics/Azizi Lab, Columbia University.

Toontown House CT Edition

Bilinear attention networks for visual question answering

Multispectral Object Detection with Yolov5

ServiceX Transformer that converts flat ROOT ntuples into columnwise data

S-attack library. Official implementation of two papers "Are socially-aware trajectory prediction models really socially-aware?" and "Vehicle trajectory prediction works, but not everywhere".

LocUNet is a deep learning method to localize a UE based solely on the reported signal strengths from a set of BSs.

OpenMMLab's Next Generation Video Understanding Toolbox and Benchmark

Official Pytorch Implementation of Adversarial Instance Augmentation for Building Change Detection in Remote Sensing Images.

Go from graph data to a secure and interactive visual graph app in 15 minutes. Batteries-included self-hosting of graph data apps with Streamlit, Graphistry, RAPIDS, and more!

This repository contains code for the paper "Disentangling Label Distribution for Long-tailed Visual Recognition", published at CVPR' 2021

FID calculation with proper image resizing and quantization steps

DeepLab-ResNet rebuilt in TensorFlow