How to detect with my model in TRT_object_detection

I succeeded by following the TRT_object_detection script as it is. The ssd_mobilenet_v2_coco model was used here. I want to run the script with the model I retrained. I means I want to use a model that I’ve re-trained with my data. I don’t know how. I have “frozen_inference_graph.pb” file.

I use Jepack 4.2, tensorflow version=1.14.0, TensorRT version=5.0

Hi,

Does your model also use ssd_mobilenet_v2 architecture?
If yes, you just need to update the file path here:

Thanks.