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):
|