tensorflow/tensorrt

Tensorflow-yolov3 from .ckpt or .pb to tf-trt model.

Closed this issue · 0 comments

Hi There,
I have trained my yolov3 model in Tensorflow, now I want to optimize it using TF-TRT but after so many tries still failed. There is too much confusing information.
I have both save model (checkpoints) and converted .pb weights.

1- I have installed tf 2.2.0 and I am traying on GTX 2080 Ti. Do i still need to install tf-trt?
2- How I can optimize these weights with tf-trt? what is the exact code?