init project
Browse files- modules/pe3r/models.py +1 -1
modules/pe3r/models.py
CHANGED
|
@@ -18,7 +18,7 @@ class Models:
|
|
| 18 |
|
| 19 |
# -- sam2 --
|
| 20 |
# SAM2_CKP = "./checkpoints/sam2.1_hiera_large.pt"
|
| 21 |
-
SAM2_CKP = '
|
| 22 |
SAM2_CONFIG = "./configs/sam2.1/sam2.1_hiera_l.yaml"
|
| 23 |
self.sam2 = build_sam2_video_predictor(SAM2_CONFIG, SAM2_CKP, device=device, apply_postprocessing=False)
|
| 24 |
self.sam2.eval()
|
|
|
|
| 18 |
|
| 19 |
# -- sam2 --
|
| 20 |
# SAM2_CKP = "./checkpoints/sam2.1_hiera_large.pt"
|
| 21 |
+
SAM2_CKP = 'hujiecpp/sam2-1-hiera-large'
|
| 22 |
SAM2_CONFIG = "./configs/sam2.1/sam2.1_hiera_l.yaml"
|
| 23 |
self.sam2 = build_sam2_video_predictor(SAM2_CONFIG, SAM2_CKP, device=device, apply_postprocessing=False)
|
| 24 |
self.sam2.eval()
|