eduardo4547 commited on
Commit
263e85a
·
verified ·
1 Parent(s): 950e1e0

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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 = []