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:
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.
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: I usepython convert_trt.py --weights ./checkpoints/rosto-porta.tf --quantize_mode float16 --output ./checkpoints/rosto-porta-fp16-trt
to converto to tensor RT format.