Spaces:
Runtime error
Runtime error
Jingkang Yang
commited on
Commit
·
d7a8f62
1
Parent(s):
97e7ba4
update: support scannet
Browse files
open_vocab_seg/utils/predictor.py
CHANGED
@@ -762,8 +762,8 @@ class VisualizationDemoIndoor(VisualizationDemo):
|
|
762 |
frame = (255. * result.detach().cpu().squeeze().permute(1, 2, 0).numpy()).astype(np.uint8)
|
763 |
frames.append(frame)
|
764 |
|
765 |
-
video_out_file = '{}.gif'.format(name)
|
766 |
-
imageio.mimwrite(os.path.join('outputs', video_out_file), frames, fps=25)
|
767 |
|
768 |
video_out_file = '{}.mp4'.format(name)
|
769 |
imageio.mimwrite(os.path.join('outputs', video_out_file), frames, fps=25, quality=8)
|
|
|
762 |
frame = (255. * result.detach().cpu().squeeze().permute(1, 2, 0).numpy()).astype(np.uint8)
|
763 |
frames.append(frame)
|
764 |
|
765 |
+
# video_out_file = '{}.gif'.format(name)
|
766 |
+
# imageio.mimwrite(os.path.join('outputs', video_out_file), frames, fps=25)
|
767 |
|
768 |
video_out_file = '{}.mp4'.format(name)
|
769 |
imageio.mimwrite(os.path.join('outputs', video_out_file), frames, fps=25, quality=8)
|