Tflite Conversion Issue with Scale

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

Tflite Conversion Issue with Scale #385

Open khg2478 opened 3 years ago

khg2478 commented 3 years ago

Hello,

I have successfully converted darknet pretrained model into tflite model using tensorflow 2.3.0 or rc version and opencv 4.1.1.26. Then, I ran detect.py but the scales of bounding boxes do not match. Any suggestions? I have fixed configuration paramters for anchor sizes, input sizes and all I believe. If I have missed anything, please let me know.

darknet tflite

KuoEuran commented 2 years ago

Hi, I think you need to go core/config.py to change the anchor box setting.