Update app.py

app.py

@@ -57,14 +57,14 @@ def stitch_rgbd_videos(
                     cmap = matplotlib.colormaps.get_cmap("inferno")
                     depth_color = (cmap(depth_norm / 255.0)[..., :3] * 255).astype(np.uint8)
                     
-                    #if len(depth_color.shape) == 3 and depth_color.shape[2] in [3, 4]:
-                    #    depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGB2GRAY)
-                    #else:
-                    #    depth_gray = depth_color                    
-                    #depth_vis = np.stack([depth_gray] * 3, axis=-1)
-
-                    depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGB2GRAY)
+                    if len(depth_color.shape) == 3 and depth_color.shape[2] in [3, 4]:
+                        depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGB2GRAY)
+                    else:
+                        depth_gray = depth_color                    
                     depth_vis = np.stack([depth_gray] * 3, axis=-1)
+
+                    #depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGB2GRAY)
+                    #depth_vis = np.stack([depth_gray] * 3, axis=-1)
                 else:
                     # Directly generate a grayscale image from the normalized depth values.
                     depth_vis = np.stack([depth_norm] * 3, axis=-1)