freealise commited on
Commit
f28cd01
1 Parent(s): 0bbf22f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -111,7 +111,8 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
111
  depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
112
 
113
  depth = depth.cpu().numpy().astype(np.uint8)
114
- depth_color = cv2.cvtColor(depth, cv2.COLOR_RGBA2GRAY)
 
115
  depth_color = cv2.cvtColor(depth_color, cv2.COLOR_GRAY2RGB)
116
 
117
  # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
 
111
  depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
112
 
113
  depth = depth.cpu().numpy().astype(np.uint8)
114
+ depth_color = cv2.applyColorMap(depth, cv2.COLORMAP_BONE)
115
+ depth_color = cv2.cvtColor(depth_color, cv2.COLOR_RGBA2GRAY)
116
  depth_color = cv2.cvtColor(depth_color, cv2.COLOR_GRAY2RGB)
117
 
118
  # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255