Spaces:
Runtime error
Runtime error
Jingkang Yang
commited on
Commit
·
46a52de
1
Parent(s):
aaf4c51
update: app
Browse files
open_vocab_seg/utils/predictor.py
CHANGED
|
@@ -179,7 +179,7 @@ class VisualizationDemo(object):
|
|
| 179 |
if "sem_seg" in predictions:
|
| 180 |
r = predictions["sem_seg"]
|
| 181 |
pred_mask = r.argmax(dim=0).to('cpu')
|
| 182 |
-
pred_mask = np.array(pred_mask, dtype=
|
| 183 |
|
| 184 |
vis_output = visualizer.draw_sem_seg(
|
| 185 |
pred_mask
|
|
@@ -239,7 +239,7 @@ class VisualizationDemo(object):
|
|
| 239 |
if "sem_seg" in predictions:
|
| 240 |
r = predictions["sem_seg"]
|
| 241 |
pred_mask = r.argmax(dim=0).to('cpu')
|
| 242 |
-
pred_mask = np.array(pred_mask, dtype=
|
| 243 |
|
| 244 |
pred_mask_sam_rgb = pred_mask.copy()
|
| 245 |
for mask in masks_rgb:
|
|
|
|
| 179 |
if "sem_seg" in predictions:
|
| 180 |
r = predictions["sem_seg"]
|
| 181 |
pred_mask = r.argmax(dim=0).to('cpu')
|
| 182 |
+
pred_mask = np.array(pred_mask, dtype=int)
|
| 183 |
|
| 184 |
vis_output = visualizer.draw_sem_seg(
|
| 185 |
pred_mask
|
|
|
|
| 239 |
if "sem_seg" in predictions:
|
| 240 |
r = predictions["sem_seg"]
|
| 241 |
pred_mask = r.argmax(dim=0).to('cpu')
|
| 242 |
+
pred_mask = np.array(pred_mask, dtype=int)
|
| 243 |
|
| 244 |
pred_mask_sam_rgb = pred_mask.copy()
|
| 245 |
for mask in masks_rgb:
|