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

ValueError: Shapes (1, 14, 14) and (1, 7, 7) are incompatible #372

Open KuoEuran opened 3 years ago

KuoEuran commented 3 years ago

Hi I use detect.py to detect my trained yolov4-tiny model. Here is my command: python detect.py --weights ./checkpoints/yolov4-tiny-224-int8.tflite --size 224 --model yolov4 --image ./data/person.jpg --framework tflite --tiny true My tflite file is generated by the command : python save_model.py --weights ./data/yolov4-tiny_final_224.weights --output ./checkpoints/yolov4-tiny-224 --input_size 224 --model yolov4 --tiny -framework tflite

python convert_tflite.py --weights ./checkpoints/yolov4-tiny-224 --output ./checkpoints/yolov4-tiny-224-int8.tflite --quantize_mode int8 However, I use the detect.py and it appears the error image Can anyone know about the issue? tks