Adding old-style shelf evaluation.
This commit is contained in:
@ -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
|
||||
|
||||
@ -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
2
skelda
Submodule skelda updated: deb926c6bc...0141ba0caf
Reference in New Issue
Block a user