search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.23k stars 1.24k forks source link

Tensorflow and tflite yolo models are of huge sizes #448

Open Aparajit-Garg opened 2 years ago

Aparajit-Garg commented 2 years ago

Hi, Thank you for a great repository. I am trying to execute the save_model.py file to generate tensorflow model for yolov4-416 original model and then converting it to tflite model using convert_tflite.py, all the programs execute successfully but the tflite model is of 245 Mb in space, which does not seem right. Can anyone please help out?

Command for pb conversion: python save_model.py --weights ./data/yolov4.weights --output ./checkpoints/yolov4-416 --input_size 416 --model yolov4 --framework tflite

Command for tflite conversion: python convert_tflite.py --weights ./checkpoints/yolov4-416 --output ./checkpoints/yolov4-416.tflite

Python version: 3.7.0 Tensorflow version: 2.3.1 (CPU) Opencv: 4.1.1.26 System: i5 11th gen Windows 10 CPU Thank you.

Duckypu commented 1 year ago

same issue :(