quoccuongle/Grounded-SAM-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: config paths in demo.py

Browse Source 
Change the config path from `configs/sam` --> `configs/samurai`
This commit is contained in:
timHau
2024-11-23 16:14:26 +01:00
parent 4b12712a1e
commit 2f0cdf0c0d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/demo.py
View File

@@ -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!")