Spaces:
Running
on
Zero
Running
on
Zero
debug
Browse files
examples/blobctrl/blobctrl_app.py
CHANGED
|
@@ -1031,7 +1031,12 @@ def segmentation(img, sel_pix):
|
|
| 1031 |
with torch.no_grad():
|
| 1032 |
masks, _, _ = mobile_predictor.predict(point_coords=np.array(points), point_labels=np.array(labels), multimask_output=False)
|
| 1033 |
print("================")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1034 |
print(masks)
|
|
|
|
| 1035 |
print("================")
|
| 1036 |
output_mask = np.ones((masks.shape[1], masks.shape[2], 3))*255
|
| 1037 |
for i in range(3):
|
|
|
|
| 1031 |
with torch.no_grad():
|
| 1032 |
masks, _, _ = mobile_predictor.predict(point_coords=np.array(points), point_labels=np.array(labels), multimask_output=False)
|
| 1033 |
print("================")
|
| 1034 |
+
print(img)
|
| 1035 |
+
print(img.shape)
|
| 1036 |
+
print(points)
|
| 1037 |
+
print(labels)
|
| 1038 |
print(masks)
|
| 1039 |
+
print(np.unique(masks))
|
| 1040 |
print("================")
|
| 1041 |
output_mask = np.ones((masks.shape[1], masks.shape[2], 3))*255
|
| 1042 |
for i in range(3):
|