Description
Hello, I have trained a custom Tensorflow-Yolov3 model.
How it is custom?
The custom means it’s not standard Yolov3 model, it’s for the two inputs (visual image and infrared image), then perform feature extraction and feature fusion and finally person object detection. My project is based on This Yolov3 TensorFlow implementation.
Now I can train, test, and use models in my system. I have both .ckpt and .pb weights. My ultimate task is to use these models in Xavier NX. I have tested the model in Xavier NX and it’s about 80% slower in NX. So I want to convert these model to trt engine and then use in Xavier NX using TensorRT. I have tried for several days but still not succeed.
There are some examples but they all use yolov3.weights and yolov3.cfg, I trained in TensorFlow and my model structure is different so I can not use cfg or weights.
So my questions are:
1- What steps I should follow to convert .ckpt or .pb weights to .trt ?
2- How I can utilize TensorRT examples for my problem as they are based on yolov3 .weights and .cfg?
3- Finally How I can use this converted .trt model for interface and real-time demo or deployment?
Environment
TensorRT Version: TensorRT-5.0.2.6
GPU Type: Xavier NX
Nvidia Driver Version: Not sure
CUDA Version: 10.2
CUDNN Version: 8.0
Operating System + Version:
Python Version (if applicable): 3.6
TensorFlow Version (if applicable): 1.15
PyTorch Version (if applicable):
Baremetal or Container (if container which image + tag):