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.24k stars 1.24k forks source link

Error converting to TensorRT #313

Open jvitors23 opened 3 years ago

jvitors23 commented 3 years ago

I'm following the steps in the README to convert an yolov4 tiny model to tensor RT. I save the model in tf format using the following command: python save_model.py --weights ./data/yolov4-tiny-train_best.weights --output ./checkpoints/rosto-porta.tf --input_size 416 --model yolov4 --tiny Then I run the convert_trt script I get the following error: image I use python convert_trt.py --weights ./checkpoints/rosto-porta.tf --quantize_mode float16 --output ./checkpoints/rosto-porta-fp16-trt to converto to tensor RT format.