Evaluated chi3d dataset.

scripts/test_skelda_dataset.py

@@ -23,6 +23,7 @@ dataset_use = "human36m"
 # dataset_use = "shelf"
 # dataset_use = "campus"
 # dataset_use = "ikeaasm"
+# dataset_use = "chi3d"
 # dataset_use = "tsinghua"
 # dataset_use = "human36m_wb"
 # dataset_use = "egohumans"
@@ -44,10 +45,6 @@ datasets = {
         "take_interval": 1,
         "with_depth": False,
     },
-    "ikeaasm": {
-        "path": "/datasets/ikeaasm/skelda/test.json",
-        "take_interval": 2,
-    },
     "campus": {
         "path": "/datasets/campus/skelda/test.json",
         "take_interval": 1,
@@ -56,6 +53,14 @@ datasets = {
         "path": "/datasets/shelf/skelda/test.json",
         "take_interval": 1,
     },
+    "ikeaasm": {
+        "path": "/datasets/ikeaasm/skelda/test.json",
+        "take_interval": 2,
+    },
+    "chi3d": {
+        "path": "/datasets/chi3d/skelda/all.json",
+        "take_interval": 5,
+    },
     "tsinghua": {
         "path": "/datasets/tsinghua/skelda/test.json",
         "take_interval": 3,
@@ -178,6 +183,11 @@ def load_labels(dataset: dict):
         for label in labels:
             label["bodyids"] = list(range(len(label["bodies3D"])))
 
+    elif "chi3d" in dataset:
+        labels = load_json(dataset["chi3d"]["path"])
+        labels = [lb for lb in labels if lb["setup"] == "s03"]
+        labels = [lb for i, lb in enumerate(labels) if i % 2000 < 150]
+
     elif "human36m_wb" in dataset:
         labels = load_json(dataset["human36m_wb"]["path"])
 
@@ -238,10 +248,10 @@ def main():
         "campus": 0.96,
         "shelf": 0.96,
         "ikeaasm": 0.89,
+        "chi3d": 0.94,
         "tsinghua": 0.96,
         "egohumans": 0.95,
         "human36m_wb": 0.94,
-        "koarob": 0.91,
     }
     minscore = minscores.get(dataset_use, 0.95)
     min_group_sizes = {
@@ -249,6 +259,7 @@ def main():
         # If the number of cameras is high, and the views are not occluded, use a higher value
         "panoptic": 1,
         "shelf": 2,
+        "chi3d": 1,
         "tsinghua": 2,
         "egohumans": 4,
     }