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