Update app.py
Browse files
app.py
CHANGED
|
@@ -125,13 +125,13 @@ def sam_process(input_image, checkpoint, tracking_points, trackings_input_label)
|
|
| 125 |
sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
|
| 126 |
model_cfg = "sam2_hiera_t.yaml"
|
| 127 |
elif checkpoint == "samll":
|
| 128 |
-
sam2_checkpoint = "./checkpoints/
|
| 129 |
model_cfg = "sam2_hiera_s.yaml"
|
| 130 |
elif checkpoint == "base-plus":
|
| 131 |
-
sam2_checkpoint = "./checkpoints/
|
| 132 |
model_cfg = "sam2_hiera_b+.yaml"
|
| 133 |
elif checkpoint == "large":
|
| 134 |
-
sam2_checkpoint = "./checkpoints/
|
| 135 |
model_cfg = "sam2_hiera_l.yaml"
|
| 136 |
|
| 137 |
sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cuda")
|
|
|
|
| 125 |
sam2_checkpoint = "./checkpoints/sam2_hiera_tiny.pt"
|
| 126 |
model_cfg = "sam2_hiera_t.yaml"
|
| 127 |
elif checkpoint == "samll":
|
| 128 |
+
sam2_checkpoint = "./checkpoints/sam2_hiera_small.pt"
|
| 129 |
model_cfg = "sam2_hiera_s.yaml"
|
| 130 |
elif checkpoint == "base-plus":
|
| 131 |
+
sam2_checkpoint = "./checkpoints/sam2_hiera_base_plus.pt"
|
| 132 |
model_cfg = "sam2_hiera_b+.yaml"
|
| 133 |
elif checkpoint == "large":
|
| 134 |
+
sam2_checkpoint = "./checkpoints/sam2_hiera_large.pt"
|
| 135 |
model_cfg = "sam2_hiera_l.yaml"
|
| 136 |
|
| 137 |
sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cuda")
|