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,
|