Update app.py

app.py

@@ -99,11 +99,11 @@ def process_image(image):
                 mask_info_list.append(mask_info)
 
                 # Asignar colores a los píxeles correspondientes en la imagen de máscaras
-                for i in range(3):
-                    mask_image[:, :, i][unique_mask] = color_bgr_255[i]
+                # Utilizamos broadcasting para asignar el color a los píxeles únicos
+                mask_image[unique_mask] = color_bgr_255
 
     # Superponer la imagen de máscaras sobre la imagen original
-    alpha = 0.4  # Transparencia ajustada (puedes ajustar este valor según tus preferencias)
+    alpha = 0.3  # Transparencia ajustada (puedes ajustar este valor según tus preferencias)
     img_with_masks = cv2.addWeighted(img_bgr.astype(np.float32), 1, mask_image.astype(np.float32), alpha, 0).astype(np.uint8)
 
     # Convertir la imagen de BGR a RGB para matplotlib