tensorflow-yolov4-tflite icon indicating copy to clipboard operation
tensorflow-yolov4-tflite copied to clipboard

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

Open KuoEuran opened this issue 4 years ago • 0 comments

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

KuoEuran avatar Jul 06 '21 09:07 KuoEuran