Spaces:
Running on Zero
Running on Zero
Upload app.py
Browse files
app.py
CHANGED
|
@@ -140,6 +140,11 @@ def segment_for_backend(image_np: np.ndarray):
|
|
| 140 |
all_masks_raw = resultado.get("masks", [])
|
| 141 |
masks_bool = [np.array(m).astype(bool) for m in all_masks_raw]
|
| 142 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 143 |
# Label map: cada pixel contiene el indice de la mascara (1-based, max 254)
|
| 144 |
label_map = np.zeros((h, w), dtype=np.uint8)
|
| 145 |
masks_out = []
|
|
|
|
| 140 |
all_masks_raw = resultado.get("masks", [])
|
| 141 |
masks_bool = [np.array(m).astype(bool) for m in all_masks_raw]
|
| 142 |
|
| 143 |
+
# Ordenar de mayor a menor area: las mascaras grandes se escriben primero
|
| 144 |
+
# y las pequenas (ventanas, detalles) las sobreescriben — evita que el muro
|
| 145 |
+
# tape a la ventana en el label_map.
|
| 146 |
+
masks_bool = sorted(masks_bool, key=lambda m: m.sum(), reverse=True)
|
| 147 |
+
|
| 148 |
# Label map: cada pixel contiene el indice de la mascara (1-based, max 254)
|
| 149 |
label_map = np.zeros((h, w), dtype=np.uint8)
|
| 150 |
masks_out = []
|