Update run_sam.py

mks0601 · Dec 17, 2024 · 43e9fb9 · 43e9fb9
1 parent 8b32157
commit 43e9fb9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/fitting/tools/code_to_copy/run_sam.py b/fitting/tools/code_to_copy/run_sam.py
@@ -40,6 +40,7 @@ def parse_args():
 # load SAM 
 ckpt_path = './sam_vit_h_4b8939.pth'
 model_type = "vit_h"
+assert osp.isfile(ckpt_path), 'Please download sam_vit_h_4b8939.pth'
 sam = sam_model_registry[model_type](checkpoint=ckpt_path).cuda()
 predictor = SamPredictor(sam)