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