yolov4-deepsort icon indicating copy to clipboard operation
yolov4-deepsort copied to clipboard

Convert_trt

Open CODEDISON opened this issue 5 years ago • 2 comments

Hi,How can I convert the tf model to tensorrt?

CODEDISON avatar Oct 08 '20 07:10 CODEDISON

@CODEDISON That's a bit difficult problem, I am also working on a similar problem. May I ask which tf model you want to convert to trt?

The better option is convert tensorflow model .ckpt > frozen_graph.pb , then convert .pb to .onnx using (https://github.com/onnx/tensorflow-onnx) and then convert .onnx to .trt using (https://github.com/onnx/onnx-tensorrt)

MuhammadAsadJaved avatar Oct 10 '20 07:10 MuhammadAsadJaved

Hi,How can I convert the tf model to tensorrt?

You can try this: https://github.com/molimat/YOLOv4-Counter-in-TF#convert-to-tensorrt

aparico avatar Dec 15 '20 02:12 aparico