Spaces:
Runtime error
Runtime error
jens commited on
Commit ·
aa4b380
1
Parent(s): 1758fb9
test lbls SAM
Browse files- inference.py +1 -0
inference.py
CHANGED
|
@@ -135,6 +135,7 @@ class SegmentPredictor:
|
|
| 135 |
self.conditioned_pred.set_image(image)
|
| 136 |
|
| 137 |
def cond_pred(self, pts, lbls):
|
|
|
|
| 138 |
masks, _, _ = self.conditioned_pred.predict(
|
| 139 |
point_coords=pts,
|
| 140 |
point_labels=lbls,
|
|
|
|
| 135 |
self.conditioned_pred.set_image(image)
|
| 136 |
|
| 137 |
def cond_pred(self, pts, lbls):
|
| 138 |
+
print(lbls)
|
| 139 |
masks, _, _ = self.conditioned_pred.predict(
|
| 140 |
point_coords=pts,
|
| 141 |
point_labels=lbls,
|