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

Running detect with tflite #289

Closed parthjdoshi closed 3 years ago

parthjdoshi commented 3 years ago

Hello. Thanks for the great repo. I had a query about running detect.py with the YOLOv4 model using the tflite framework. While the core YOLOv4 file contains a decode_tflite function, why is it not called when running detect.py? How can the results of detect.py be directly passed to the filter_boxes function without postprocessing the output?

Once again, thank you for your help.

zero90169 commented 3 years ago

There is a FLAG in decode function called FRAMEWORK, default setting is 'tf'. When we inference the tflite file the FRAMEWORK FLAG will be override into 'tflite' and the decode function will use decode_tflite function.

parthjdoshi commented 3 years ago

Oh yes. I missed that. Thank you :)