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,
|
"correct_parts": 6623,
|
||||||
"pcp": 0.991764
|
"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
|
### Campus
|
||||||
|
|||||||
@ -383,6 +383,19 @@ def main():
|
|||||||
replace_head_with_nose=True,
|
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__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
2
skelda
2
skelda
Submodule skelda updated: deb926c6bc...0141ba0caf
Reference in New Issue
Block a user