Jingkang Yang commited on
Commit
d7a8f62
1 Parent(s): 97e7ba4

update: support scannet

Browse files
Files changed (1) hide show
  1. open_vocab_seg/utils/predictor.py +2 -2
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)