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.24k stars 1.24k forks source link

Converter not working for custom object trained model #311

Open alejandroariaszuluaga opened 3 years ago

alejandroariaszuluaga commented 3 years ago

Hey guys, a bit new here, would like to know if I'm doing something wrong. Wanted to convert my custom trained YoloV4-tiny model (for 1 specific class) to tflite. But first wanted to convert it to a Tensorflow SavedModel

python save_model.py --weights {route to my .weights file} --output ./checkpoints/yolov4-416 --input_size 416 --model yolov4 --framework tflite

Also would like to understand what the --output ./checkpoints/yolov4-416 does, maybe my error is there.

Thanks in advance!

wooruang commented 3 years ago

If you want save a model for tiny, you should put --tiny in argument.