Adding old-style shelf evaluation.

This commit is contained in:
Daniel
2025-05-26 13:57:50 +02:00
parent dbc92787dd
commit b8c5159588
3 changed files with 34 additions and 1 deletions

View File

@ -575,6 +575,26 @@ Results of the model in various experiments on different datasets. \
"correct_parts": 6623,
"pcp": 0.991764
}
{
"total_parts": {
"1": 2790,
"3": 1610,
"2": 370,
"sum": 4770
},
"correct_parts": {
"1": 2783,
"3": 1574,
"2": 351,
"sum": 4708
},
"pcp": {
"1": 0.997491,
"3": 0.97764,
"2": 0.948649,
"avg": 0.974593
}
}
```
### Campus

View File

@ -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__":

2
skelda

Submodule skelda updated: deb926c6bc...0141ba0caf