dillonlaird commited on
Commit
153ee9b
·
1 Parent(s): 80c2e04

fixed name error

Browse files
Files changed (1) hide show
  1. 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.postprocessing import clean_mask_torch
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):