RuoyuFeng commited on
Commit
8732e6b
1 Parent(s): 82c08d7

back to stable

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -50,8 +50,7 @@ def get_masked_img(img, w, h, features, orig_h, orig_w, input_h, input_w):
50
  # model['sam'].input_size = input_size
51
  # model['sam'].is_image_set = True
52
 
53
- # model['sam'].set_image(img)
54
- # masks, _, _ = predictor.predict(
55
  masks, _, _ = model['sam'].predict(
56
  point_coords=np.array([point_coords]),
57
  point_labels=np.array(point_labels),
 
50
  # model['sam'].input_size = input_size
51
  # model['sam'].is_image_set = True
52
 
53
+ model['sam'].set_image(img)
 
54
  masks, _, _ = model['sam'].predict(
55
  point_coords=np.array([point_coords]),
56
  point_labels=np.array(point_labels),