Adding old-style shelf evaluation.

2025-05-26 13:57:50 +02:00
parent dbc92787dd
commit b8c5159588
3 changed files with 34 additions and 1 deletions
--- a/scripts/test_skelda_dataset.py
+++ b/scripts/test_skelda_dataset.py
@ -383,6 +383,19 @@ def main():
        replace_head_with_nose=True,
    )

+    if dataset_use == "shelf":
+        # Also run old-style evaluation for shelf dataset
+        odir = os.path.join(output_dir, "pcp/") if output_dir != "" else ""
+        _ = evals.campus_shelf.run_eval(
+            labels,
+            all_poses_3d,
+            all_ids,
+            joint_names_net=joint_names_3d,
+            save_error_imgs=odir,
+            debug_2D_preds=all_poses_2d,
+        )
+
+
 # ==================================================================================================

 if __name__ == "__main__":