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

Merge pull request #11 from timHau/master

Browse Source 
Fix: config paths in demo.py
This commit is contained in:
Cheng-Yen Yang
2024-11-23 10:31:50 -08:00
committed by GitHub
parent 4b12712a1e 2f0cdf0c0d
commit 5b31709e2c
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): def determine_model_cfg(model_path):
if "large" in 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: 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: 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: elif "tiny" in model_path:
return "configs/sam2.1/sam2.1_hiera_t.yaml" return "configs/samurai/sam2.1_hiera_t.yaml"
else: else:
raise ValueError("Unknown model size in path!") raise ValueError("Unknown model size in path!")