search

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

Unsupported data type encountered in input 0 tensorRT int8 quantize mode yolov3 tiny #374

Open kaijaz123 opened 3 years ago

kaijaz123 commented 3 years ago

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