Load ./libmmdeploy_tensorrt_ops.so for TensorRT EP if available

rtmo_gpu.py

@@ -3,6 +3,7 @@ import numpy as np
 from typing import List, Tuple
 import onnxruntime as ort
 import cv2
+os.environ['ORT_TENSORRT_EXTRA_PLUGIN_LIB_PATHS']='libmmdeploy_tensorrt_ops.so'
 
 # dictionary from https://github.com/Tau-J/rtmlib/blob/4b29101d54b611048ef165277cebfffff3030074/rtmlib/visualization/skeleton/coco17.py
 coco17 = dict(name='coco17',
@@ -486,4 +487,4 @@ class RTMO_GPU_Batch(RTMO_GPU):
         batch_img, ratios = self.preprocess_batch(images)
         outputs = self.inference(batch_img)
         keypoints, scores = self.postprocess_batch(outputs, ratios)
-        return keypoints, scores
+        return keypoints, scores