Hugging Face's logo

Spaces:
jotase
/
florence-sam-masking
Running on Zero

App Files Community
jotase commited on
Commit
7cbb253
·
verified ·
1 Parent(s): baecb28

Update utils/sam.py

Browse files
Files changed (1) hide show
  1. utils/sam.py +1 -1
utils/sam.py CHANGED
@@ -38,7 +38,7 @@ def run_sam_inference(
38
  image = np.array(image.convert("RGB"))
39
  model.set_image(image)
40
  mask, score, _ = model.predict(box=detections.xyxy, multimask_output=False)
41
-
42
  # dirty fix; remove this later
43
  print("Mask Shape lenght: ", len(mask.shape))
44
  if len(mask.shape) == 4:
 
38
  image = np.array(image.convert("RGB"))
39
  model.set_image(image)
40
  mask, score, _ = model.predict(box=detections.xyxy, multimask_output=False)
41
+ print("Prediction score: ", score)
42
  # dirty fix; remove this later
43
  print("Mask Shape lenght: ", len(mask.shape))
44
  if len(mask.shape) == 4: