Skip to content

Commit

Permalink
merge changes regarding thresholds
Browse files Browse the repository at this point in the history
Signed-off-by: Martin <[email protected]>
  • Loading branch information
bmmtstb committed Jun 6, 2024
1 parent e8feef4 commit 3a8a149
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions scripts/own/eval_single_similarity_const.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ def run(config: Config, dl_key: str, paths: list[str], out_key: str) -> None:
cfg = load_config(CONFIG_FILE)
base_path = cfg["dgs_gt"]["base_path"]
data_paths = [f.path for f in os.scandir(base_path) if f.is_file()]
run(config=cfg, dl_key="dgs_gt", paths=data_paths)
run(config=cfg, dl_key="dgs_gt", paths=data_paths, out_key="dgs_gt")

print("Evaluating on the PT21 eval-dataset using KeypointRCNN as prediction backbone")
for thresh in (pbar_thresh := tqdm(["085", "090", "095", "099"], desc="thresholds")):
Expand All @@ -109,4 +109,4 @@ def run(config: Config, dl_key: str, paths: list[str], out_key: str) -> None:
cfg["dgs_rcnn"]["base_path"] = base_path
cfg["dgs_rcnn"]["crops_folder"] = f"./data/PoseTrack21/crops/256x192/rcnn_prediction_{thresh}/"
data_paths = [f.path for f in os.scandir(base_path) if f.is_file()]
run(config=cfg, dl_key="dgs_rcnn", paths=data_paths)
run(config=cfg, dl_key="dgs_rcnn", paths=data_paths, out_key=f"dgs_rcnn_{thresh}")

0 comments on commit 3a8a149

Please sign in to comment.