Update controlnet_module.py

controlnet_module.py

@@ -305,11 +305,22 @@ class ControlNetProcessor:
                 print(f"working_mask shape: {working_mask.shape}")
 
 
-                # DILATE auf der weißen Person - daduch wird Person etwas vergrößert
-                kernel_dilate = np.ones((5, 5), np.uint8)               
-                working_mask = cv2.dilate(working_mask, kernel_dilate, iterations=1)
-                print(f"   ✅ Dilate (5x5) - Person leicht erweitert")
-
+                # === 1. FÜR INPAINT: Kleinere Dilatation (5×5) === 
+                #Inpaint darf auch nur eine schwarz/weiß Maske (Binärmaske 0,255) bekommen. Bei
+                #weichen/grauen Pixeln könnte Inpaint etwas anderes generieren. Inpaint darf auch nur kleinere
+                #Segmentvergrößerung erhalten während Compositing größeren Dilate - also mehr weiße Pixel um Person benötigt
+                inpaint_mask = working_mask.copy()  # Person=weiß
+
+                kernel_inpaint = np.ones((5, 5), np.uint8)
+                inpaint_mask = cv2.dilate(inpaint_mask, kernel_inpaint, iterations=1)
+                print(f"   ✅ Inpaint Dilate (5x5) - Person leicht erweitert")
+                # Später invertieren für Inpaint
+
+                # === 2. FÜR COMPOSITING: Größere Dilatation (15×15) auf working_mask ===
+                kernel_comp = np.ones((15, 15), np.uint8)
+                working_mask = cv2.dilate(working_mask, kernel_comp, iterations=1)
+                print(f"   ✅ Compositing Dilate (15x15) - Person stark erweitert")
+             
 
                 # MORPH_CLOSE auf dem schwarzen Hintergrund (feine Löcher)- kleiner Kernel filigrane Heranarbeitung an Person,
                 # es werden aber auch nur kleine Löcher in Umgebung von weiß nach schwarz geändert!
@@ -344,12 +355,10 @@ class ControlNetProcessor:
                     print(f"   ✅ Konturenfilter - Größte Kontur behalten, {len(contours)-1} kleine entfernt")
 
 
-                #Inpaint darf nur eine schwarz/weiß Maske (Binärmaske 0,255) bekommen. Bei
-                #weichen/grauen Pixeln könnte Inpaint etwas anderes generieren. Die folgende
-                #Blur-Maske für Composition und UI-Anzeige
+      
                 inpaint_binary_mask = working_mask.copy()
                 #Invertieren da SAM segmentierte Person/Gegenstand weiß ausgibt
-                inpaint_binary_mask = 255 - inpaint_binary_mask
+                inpaint_binary_mask = 255 - inpaint_mask
                 
 
                 # Gaussian-BLUR für weiche Kanten für Composition