Skip to content

Convert MobileNetV3Small defined and pre-trained in PyTorch to a TFLite quantized model

Notifications You must be signed in to change notification settings

mgarbade/pytorch-to-tflite-example

 
 

Repository files navigation

pytorch-to-tflite-example

Convert MobileNetV3Small defined and pre-trained in PyTorch to a TFLite quantized model

Requirements

Python >= 3.6.0

Python packages:

  • Keras==2.2.4
  • onnx==1.5.0
  • onnx2keras==0.0.3
  • tensorflow==1.14.0
  • torch==1.1.0
  • Pillow==6.1.0

Usage

Download weights

Download manually from here
or use

./download_weight.sh

Run the script

python3 main.py

About

Convert MobileNetV3Small defined and pre-trained in PyTorch to a TFLite quantized model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.0%
  • Shell 1.0%