Convert custom trained yolov4 to tflite

https://github.com/hunglc007/tensorflow-yolov4-tflite

change xxxx.names at this config file:

https://github.com/hunglc007/tensorflow-yolov4-tflite/blob/master/core/config.py#L14

Code

conda activate yolo-tflite

(yolov4-tflite environment is for another folder: Keras-..., not working)

python save_model.py --weights ./data/yolov4-drone_final.weights --output ./checkpoints/yolo_drone --input_size 416 --model yolov4

python detect.py --weights ./checkpoints/yolo_drone --size 416 --model yolov4 --image ./data/drone.jpg

 

-------------------------------------------------------------------

restart:

#only tf model

python save_model.py --weights ./data/yolov4-drone_final.weights --output ./checkpoints/yolo_drone --input_size 576 --model yolov4 --framework tflite

 

#tf to tflite

python convert_tflite.py --weights ./checkpoints/yolo_drone --output ./checkpoints/yolo_drone.tflite

 

#run tflite demo

python detect.py --weights ./checkpoints/yolo_drone.tflite --size 576 --model yolov4 --image ./data/drone.jpg --framework tflite

 

#online tflite viewer website:

https://lutzroeder.github.io/netron/