Spaces:
Running
Running
Update app.py
Browse files
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.
|
|
|
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
|