From 2f0cdf0c0d5a5ecfdef3118915e10995ee109b92 Mon Sep 17 00:00:00 2001 From: timHau Date: Sat, 23 Nov 2024 16:14:26 +0100 Subject: [PATCH] Fix: config paths in demo.py Change the config path from `configs/sam` --> `configs/samurai` --- scripts/demo.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/scripts/demo.py b/scripts/demo.py index e6b39fc..e2ded94 100644 --- a/scripts/demo.py +++ b/scripts/demo.py @@ -23,13 +23,13 @@ def load_txt(gt_path): def determine_model_cfg(model_path): if "large" in model_path: - return "configs/sam2.1/sam2.1_hiera_l.yaml" + return "configs/samurai/sam2.1_hiera_l.yaml" elif "base_plus" in model_path: - return "configs/sam2.1/sam2.1_hiera_b+.yaml" + return "configs/samurai/sam2.1_hiera_b+.yaml" elif "small" in model_path: - return "configs/sam2.1/sam2.1_hiera_s.yaml" + return "configs/samurai/sam2.1_hiera_s.yaml" elif "tiny" in model_path: - return "configs/sam2.1/sam2.1_hiera_t.yaml" + return "configs/samurai/sam2.1_hiera_t.yaml" else: raise ValueError("Unknown model size in path!")