tensorflow-yolov4-tflite
tensorflow-yolov4-tflite copied to clipboard
Unsupported data type encountered in input 0 tensorRT int8 quantize mode yolov3 tiny
When converting yolov3 tiny to tensorRT format I get this error.
tensorflow.python.framework.errors_impl.InvalidArgumentError: Unsupported data type encountered in input 0
[[node TRTEngineOp_0_0 (defined at convert_trt.py:56) ]] [Op:__inference_pruned_11564]
Command to run:
!python save_model.py --weights yolov3-tiny_10000.weights --output ./checkpoints/yolov3-tiny-ball.tf --input_size 416 --model yolov3 --tiny
!python convert_trt.py --weights ./checkpoints/yolov3-tiny-ball.tf --quantize_mode int8 --output ./checkpoints/v3-tiny-ball-416-trt-int8
tensorflow version : 2.3.0rc0