Spaces:
Runtime error
Runtime error
jens
commited on
Commit
·
fea2110
1
Parent(s):
aa4b380
cast to array
Browse files- inference.py +5 -0
inference.py
CHANGED
|
@@ -135,7 +135,12 @@ class SegmentPredictor:
|
|
| 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,
|
|
|
|
| 135 |
self.conditioned_pred.set_image(image)
|
| 136 |
|
| 137 |
def cond_pred(self, pts, lbls):
|
| 138 |
+
lbls = np.array(lbls)
|
| 139 |
+
pts = np.array(pts)
|
| 140 |
+
|
| 141 |
+
print(pts)
|
| 142 |
print(lbls)
|
| 143 |
+
|
| 144 |
masks, _, _ = self.conditioned_pred.predict(
|
| 145 |
point_coords=pts,
|
| 146 |
point_labels=lbls,
|