Converted wb model as well.

scripts/test_skelda_dataset.py

@@ -116,6 +116,7 @@ datasets = {
         "path": "/datasets/human36m/skelda/wb/test.json",
         "take_interval": 100,
         "min_bbox_score": 0.4,
+        "batch_poses": False,
     },
     "egohumans_tagging": {
         "path": "/datasets/egohumans/skelda/all.json",

scripts/utils_2d_pose_ort.py

@@ -479,12 +479,18 @@ def load_model(min_bbox_score=0.3, min_bbox_area=0.1 * 0.1, batch_poses=False):
     return model
 
 
-def load_wb_model():
-    print("Loading mmpose whole body model ...")
+def load_wb_model(min_bbox_score=0.3, min_bbox_area=0.1 * 0.1, batch_poses=False):
+    print("Loading 2D-WB model ...")
 
-    model = None
+    model = TopDown(
+        "/RapidPoseTriangulation/extras/mmdeploy/exports/rtmdet-nano_1x320x320x3_fp16_extra-steps.onnx",
+        f"/RapidPoseTriangulation/extras/mmdeploy/exports/rtmpose-l_wb_{'B' if batch_poses else '1'}x384x288x3_fp16_extra-steps.onnx",
+        box_conf_threshold=min_bbox_score,
+        box_min_area=min_bbox_area,
+        warmup=30,
+    )
 
-    print("Loaded mmpose model")
+    print("Loaded 2D-WB model")
     return model