Update utils/dc_utils.py

utils/dc_utils.py

@@ -71,7 +71,7 @@ def read_video_frames(video_path, process_length, target_fps=-1, max_res=-1):
     return frames, fps
 
 
-def save_video(frames, output_video_path, fps=10, is_depths=False):
+def save_video(frames, output_video_path, fps=10, is_depths=False, grayscale=False):
     writer = imageio.get_writer(output_video_path, fps=fps, macro_block_size=1, codec='libx264', ffmpeg_params=['-crf', '18'])
     if is_depths:
         colormap = np.array(cm.get_cmap("inferno").colors)
@@ -79,7 +79,7 @@ def save_video(frames, output_video_path, fps=10, is_depths=False):
         for i in range(frames.shape[0]):
             depth = frames[i]
             depth_norm = ((depth - d_min) / (d_max - d_min) * 255).astype(np.uint8)
-            depth_vis = (colormap[depth_norm] * 255).astype(np.uint8)
+            depth_vis = (colormap[depth_norm] * 255).astype(np.uint8) if not grayscale else depth_norm
             writer.append_data(depth_vis)
     else:
         for i in range(frames.shape[0]):