-
将PyTorch中已定义和预训练的MobileNetV3Small模型转换为TFLite量化版本的示例
资源介绍
pytorch到tflite的例子
将在PyTorch中定义和预训练的MobileNetV3Small转换为TFLite量化模型
要求
Python> = 3.6.0
Python套件：
Keras==2.2.4
onnx==1.5.0
onnx2keras==0.0.3
tensorflow==1.14.0
torch==1.1.0
Pillow==6.1.0
用法
下载砝码
./download_weight.sh
运行脚本
python3 main.py