Update app.py

app.py

@@ -88,19 +88,13 @@ def stitch_rgbd_videos(
             H_full, W_full = rgb_full.shape[:2]
             depth_vis_resized = cv2.resize(depth_vis, (W_full, H_full))
             
-            # Ensure both images have 3 channels.
-            #if len(rgb_full.shape) == 2:
-            #    rgb_full = cv2.cvtColor(rgb_full, cv2.COLOR_GRAY2BGR)
-            #if len(depth_vis_resized.shape) == 2:
-            #    depth_vis_resized = cv2.cvtColor(depth_vis_resized, cv2.COLOR_GRAY2BGR)
+            # Ensure both are 3-channel images:
+            if len(rgb_full.shape) == 3 and len(depth_vis_resized.shape) == 2:
+                depth_vis_resized = cv2.cvtColor(depth_vis_resized, cv2.COLOR_GRAY2BGR)
             
-            # Ensure same data type.
-            #if rgb_full.dtype != depth_vis_resized.dtype:
-            #    depth_vis_resized = depth_vis_resized.astype(rgb_full.dtype)
-            
-            # Ensure images are contiguous in memory.
-            #rgb_full = np.ascontiguousarray(rgb_full)
-            #depth_vis_resized = np.ascontiguousarray(depth_vis_resized)
+            # Ensure both are the same type (commonly uint8):
+            if rgb_full.dtype != depth_vis_resized.dtype:
+                depth_vis_resized = depth_vis_resized.astype(rgb_full.dtype)
             
             # Now safely concatenate.
             stitched = cv2.hconcat([rgb_full, depth_vis_resized])