Use fp32 model for whole-body poses.

2024-12-19 11:12:13 +01:00
parent 1bb7dd966e
commit 19bb1a6372
2 changed files with 6 additions and 3 deletions
--- a/scripts/test_skelda_dataset.py
+++ b/scripts/test_skelda_dataset.py
@ -190,9 +190,11 @@ eval_joints = [
    "ankle_left",
    "ankle_right",
 ]
-if dataset_use in ["human36m", "panoptic"]:
+if dataset_use == "human36m":
    eval_joints[eval_joints.index("head")] = "nose"
-if dataset_use.endswith("_wb"):
+if dataset_use == "panoptic":
+    eval_joints[eval_joints.index("head")] = "nose"
+if dataset_use == "human36m_wb":
    if any((test_triangulate.whole_body.values())):
        eval_joints = list(joint_names_2d)
    else: