s194649 commited on
Commit
bcdfff1
·
1 Parent(s): dcd8315

image set true flag

Browse files
Files changed (1) hide show
  1. inference.py +1 -0
inference.py CHANGED
@@ -253,6 +253,7 @@ class CustomSamPredictor(SamPredictor):
253
  (np.ndarray): Low resolution mask logits for subsequent iterations.
254
  """
255
  self.features = embedding
 
256
  return self.predict(
257
  point_coords=point_coords,
258
  point_labels=point_labels,
 
253
  (np.ndarray): Low resolution mask logits for subsequent iterations.
254
  """
255
  self.features = embedding
256
+ self.is_image_set = True
257
  return self.predict(
258
  point_coords=point_coords,
259
  point_labels=point_labels,