Fix trimesh index out of bounds error: Resolve dimension mismatch between image and depth map - Resize image to match depth map dimensions exactly - Add bounds checking for array access - Improve face generation with validation - Add detailed dimension logging for debugging

models/depth_processor.py

@@ -324,9 +324,15 @@ class DepthProcessor:
         try:
             import trimesh
             
-            img_array = np.array(image)
+            # Ensure image and depth map have same dimensions
             h, w = depth_map.shape
             
+            # Resize image to match depth map dimensions exactly
+            image_resized = image.resize((w, h), Image.Resampling.LANCZOS)
+            img_array = np.array(image_resized)
+            
+            logger.info(f"📏 Image dimensions: {img_array.shape}, Depth map: {depth_map.shape}")
+            
             vertices = []
             faces = []
             colors = []
@@ -343,6 +349,10 @@ class DepthProcessor:
             
             for y in range(0, h, step):
                 for x in range(0, w, step):
+                    # Ensure we don't go out of bounds
+                    if y >= h or x >= w:
+                        continue
+                        
                     depth_val = depth_map[y, x]
                     Z = (1.0 - depth_val) * 0.3  # Reduced depth scale
                     
@@ -356,15 +366,22 @@ class DepthProcessor:
                         vertex_map[(y, x)] = vertex_idx
                         vertex_idx += 1
                         
-                        # Add color
-                        if len(img_array.shape) == 3:
-                            colors.append(img_array[y, x] / 255.0)
+                        # Add color - ensure coordinates are within bounds
+                        if y < img_array.shape[0] and x < img_array.shape[1]:
+                            if len(img_array.shape) == 3:
+                                colors.append(img_array[y, x] / 255.0)
+                            else:
+                                colors.append([0.7, 0.7, 0.7])
                         else:
-                            colors.append([0.7, 0.7, 0.7])
+                            colors.append([0.7, 0.7, 0.7])  # Default gray color
             
-            # Generate faces from grid
+            # Generate faces from grid with bounds checking
             for y in range(0, h - step, step):
                 for x in range(0, w - step, step):
+                    # Ensure all corners are within bounds
+                    if y + step >= h or x + step >= w:
+                        continue
+                        
                     corners = [
                         (y, x), (y, x + step),
                         (y + step, x), (y + step, x + step)
@@ -372,8 +389,10 @@ class DepthProcessor:
                     
                     if all(corner in vertex_map for corner in corners):
                         indices = [vertex_map[corner] for corner in corners]
-                        faces.append([indices[0], indices[1], indices[2]])
-                        faces.append([indices[1], indices[3], indices[2]])
+                        # Only create faces if we have valid indices
+                        if len(indices) == 4 and all(idx is not None for idx in indices):
+                            faces.append([indices[0], indices[1], indices[2]])
+                            faces.append([indices[1], indices[3], indices[2]])
             
             if not vertices:
                 raise ValueError("No valid vertices generated")