Skip to content

Commit

Permalink
fix: wrong hb,itodd dataset name in bop eval + use pose scores
Browse files Browse the repository at this point in the history
  • Loading branch information
MedericFourmy committed Aug 24, 2023
1 parent a6e5b9c commit 693d9f2
Showing 1 changed file with 4 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -182,9 +182,9 @@ def run_evaluation(cfg: BOPEvalConfig) -> None:
results_path = Path(cfg.results_path)
eval_dir = Path(cfg.eval_dir)

if cfg.dataset == "hb" and cfg.split == "test":
if cfg.dataset == "hb.bop19" and cfg.split == "test":
cfg.convert_only = True
if cfg.dataset == "itodd" and cfg.split == "test":
if cfg.dataset == "itodd.bop19" and cfg.split == "test":
cfg.convert_only = True

scores_pose_path = None
Expand All @@ -196,7 +196,8 @@ def run_evaluation(cfg: BOPEvalConfig) -> None:
# this is what is expected by bop_toolkit_lib
csv_path = eval_dir / f"{method}_{cfg.dataset.split('.')[0]}-{cfg.split}.csv"

convert_results_to_bop(results_path, csv_path, cfg.method, use_pose_score=False)
# pose scores give better AR scores in general
convert_results_to_bop(results_path, csv_path, cfg.method, use_pose_score=True)

if not cfg.convert_only:
_run_bop_evaluation(csv_path, cfg.eval_dir, eval_detection=False)
Expand Down

0 comments on commit 693d9f2

Please sign in to comment.