Fixed running custom onnx models.

2024-11-29 15:18:57 +01:00
parent f6d13ea5a7
commit 93d4611a91
6 changed files with 158 additions and 32 deletions
--- a/scripts/utils_2d_pose_ep.py
+++ b/scripts/utils_2d_pose_ep.py
@ -14,7 +14,15 @@ filepath = os.path.dirname(os.path.realpath(__file__)) + "/"
 def load_model():
    print("Loading mmpose model ...")

-    model = ep.TopDown("rtmpose_m", "SimCC", "rtmdet_s")
+    # model = ep.TopDown("rtmpose_m", "SimCC", "rtmdet_s")
+    model = ep.TopDown(
+        "/RapidPoseTriangulation/extras/mmdeploy/exports/rtmpose-m_384.onnx",
+        "SimCC",
+        "/RapidPoseTriangulation/extras/mmdeploy/exports/rtmdet_nano_320.onnx",
+        conf_threshold=0.3,
+        iou_threshold=0.3,
+        warmup=10,
+    )

    print("Loaded mmpose model")
    return model