Spaces:
Build error
Build error
Commit
·
153ee9b
1
Parent(s):
80c2e04
fixed name error
Browse files- app/per_sam/model.py +1 -1
app/per_sam/model.py
CHANGED
@@ -8,7 +8,7 @@ import cv2
|
|
8 |
from torchvision.ops.boxes import batched_nms
|
9 |
from app.mobile_sam import SamPredictor
|
10 |
from app.mobile_sam.utils import batched_mask_to_box
|
11 |
-
from app.sam.
|
12 |
|
13 |
|
14 |
def point_selection(mask_sim, topk: int = 1):
|
|
|
8 |
from torchvision.ops.boxes import batched_nms
|
9 |
from app.mobile_sam import SamPredictor
|
10 |
from app.mobile_sam.utils import batched_mask_to_box
|
11 |
+
from app.sam.postprocess import clean_mask_torch
|
12 |
|
13 |
|
14 |
def point_selection(mask_sim, topk: int = 1):
|