Update controlnet_module.py

controlnet_module.py

@@ -868,7 +868,7 @@ class ControlNetProcessor:
                 # Heuristik und Postprocessing auf Crop
                 #=========================
                 
-                # Masken speichern in den Arrays
+                # Masken speichern in Array
                 all_masks_crop = []        #Weiterverarbeitung in Crop-Größe   
 
                 
@@ -882,7 +882,7 @@ class ControlNetProcessor:
                         align_corners=False
                     ).squeeze()
                     
-                    mask_np = resized_mask.sigmoid().cpu().numpy()
+                    mask_np = resized_mask_crop.sigmoid().cpu().numpy()
                     all_masks_crop.append(mask_np)
 
 
@@ -1139,7 +1139,8 @@ class ControlNetProcessor:
                 print("📊 FINALE MASKEN-STATISTIK")
 
                 # Weiße Pixel zählen
-                white_pixels = np.sum(mask_array > 0)
+                final_array = np.array(mask_original)  # ✅ Original-Größe
+                white_pixels = np.sum(final_array > 0)
                 total_pixels = mask_array.size
                 white_ratio = white_pixels / total_pixels * 100 if total_pixels >0 else 0
 
@@ -1165,7 +1166,7 @@ class ControlNetProcessor:
               
                 print("#" * 80)
                 print(f"✅ SAM 2 SEGMENTIERUNG ABGESCHLOSSEN")
-                print(f"📐 Finale Maskengröße: {mask_512.size}")  # Immer 512×512
+                print(f"📐 Finale Maskengröße: {mask_original.size}")
                 print(f"🎛️  Verwendeter Modus: {mode}")
                 print(f"👤 Crop={crop_size}×{crop_size}px, Heuristik-Score={best_score:.3f}")
                 print(f"👤 Kopfabdeckung: {coverage_ratio:.1%} der BBox")