Problem converting ONNX model to TensorRT Engine for SSD Mobilenet V2

AastaLLL · June 30, 2020, 2:46am

Hi,

The FlattenConcat is by default installed in the TensorRT 7.1.
So please comment the plugin library link here to skip the error:

github.com

AastaNV/TRT_object_detection/blob/master/main.py#L18


      
          import pycuda.driver as cuda
          
          
import coco
          import uff
          import tensorrt as trt
          import graphsurgeon as gs
          #from config import model_ssd_inception_v2_coco_2017_11_17 as model
          #from config import model_ssd_mobilenet_v1_coco_2018_01_28 as model
          from config import model_ssd_mobilenet_v2_coco_2018_03_29 as model
          
          
ctypes.CDLL("lib/libflattenconcat.so")
          COCO_LABELS = coco.COCO_CLASSES_LIST
          
          

          
# initialize
          TRT_LOGGER = trt.Logger(trt.Logger.INFO)
          trt.init_libnvinfer_plugins(TRT_LOGGER, '')
          runtime = trt.Runtime(TRT_LOGGER)
          
          

          
# compile model into TensorRT

Thanks.

Topic		Replies	Views
Unsupported ONNX data type: UINT8 (2) TensorRT	24	9087	May 6, 2021
I am trying to convert the ONNX SSD mobilnet v3 model into TensorRT Engine. I am getting the below error Jetson TX2 tensorrt , tensorflow	24	3804	February 17, 2022
Exporting Tensorflow models to Jetson Nano Jetson Nano tensorflow	25	6766	October 15, 2021
Convert pb to uff TensorRT tensorrt , jetson-inference	9	2592	May 18, 2021
Cannot convert SSD ONNX model to TensorRT TensorRT tensorrt	15	2395	November 23, 2022
mobilenet onnx problem TensorRT	10	1896	October 12, 2021
Tensorflow MobilenetSSD v2 conversion in TensorRT TensorRT tensorrt , tensorflow	3	1022	March 8, 2021
Convert Tensoflow V2 model ( pb format) to UFF Jetson Nano tensorflow	7	1345	October 18, 2021
Inference error while using tensorrt engine on jetson nano Jetson Nano tensorrt , nvbugs	23	3728	April 20, 2022
I am trying to convert the ONNX SSD mobilnet v2 model into TensorRT Engine. I am getting the below error Jetson AGX Xavier tensorrt , jetson	8	817	December 8, 2021

Problem converting ONNX model to TensorRT Engine for SSD Mobilenet V2

Related topics