Spaces:
Runtime error
Runtime error
s194649
commited on
Commit
·
bcdfff1
1
Parent(s):
dcd8315
image set true flag
Browse files- 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,
|