Updated ikeaasm and tsinghua dataset splits. Small other updates.

scripts/test_skelda_dataset.py

@@ -141,7 +141,8 @@ def load_labels(dataset: dict):
 
     elif "ikeaasm" in dataset:
         labels = load_json(dataset["ikeaasm"]["path"])
-        labels = [lb for i, lb in enumerate(labels) if i % 300 < 72]
+        cams0 = str(labels[0]["cameras"])
+        labels = [lb for lb in labels if str(lb["cameras"]) == cams0]
 
     elif "shelf" in dataset:
         labels = load_json(dataset["shelf"]["path"])
@@ -154,7 +155,8 @@ def load_labels(dataset: dict):
     elif "tsinghua" in dataset:
         labels = load_json(dataset["tsinghua"]["path"])
         labels = [lb for lb in labels if "test" in lb["splits"]]
-        labels = [lb for i, lb in enumerate(labels) if i % 800 < 90]
+        labels = [lb for lb in labels if lb["seq"] == "seq_1"]
+        labels = [lb for i, lb in enumerate(labels) if i % 300 < 90]
 
         for label in labels:
             label["bodyids"] = list(range(len(label["bodies3D"])))
@@ -237,6 +239,9 @@ def add_missing_joints(poses3D, joint_names_3d):
 
     for i in range(len(poses3D)):
         valid_joints = np.where(poses3D[i, :, 3] > 0.1)[0]
+
+        if len(valid_joints) == 0:
+            continue
         body_center = np.mean(poses3D[i, valid_joints, :3], axis=0)
 
         for j in range(len(joint_names_3d)):
@@ -360,7 +365,7 @@ def main():
             "mvor": 0.86,
             "campus": 0.96,
             "shelf": 0.96,
-            "ikeaasm": 0.86,
+            "ikeaasm": 0.89,
             "tsinghua": 0.96,
             "human36m_wb": 0.94,
         }
@@ -416,6 +421,7 @@ def main():
         joint_names_net=joint_names_3d,
         joint_names_use=eval_joints,
         save_error_imgs=output_dir,
+        pred_imgpaths=all_paths,
     )
     _ = evals.pcp.run_eval(
         labels,

scripts/test_triangulate.py

@@ -306,7 +306,7 @@ def filter_poses(poses3D, poses2D, roomparams, joint_names, drop_few_limbs=True)
         if average_length < 0.1:
             drop.append(i)
             continue
-        if drop_few_limbs and average_length > 0.5:
+        if total_limbs > 4 and average_length > 0.5:
             drop.append(i)
             continue