tensorflow-yolov4-tflite
tensorflow-yolov4-tflite copied to clipboard
ValueError: Shapes (1, 14, 14) and (1, 7, 7) are incompatible
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
Can anyone know about the issue? tks