Astridkraft commited on
Commit
045fdda
·
verified ·
1 Parent(s): 9a1b19b

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +7 -1
controlnet_module.py CHANGED
@@ -594,6 +594,11 @@ class ControlNetProcessor:
594
 
595
  kernel_dilate = np.ones((15, 15), np.uint8)
596
  mask_array = cv2.dilate(mask_array, kernel_dilate, iterations=1)
 
 
 
 
 
597
 
598
  # 3. Weiche Übergänge mittlerer Blur für natürliche Übergänge
599
  mask_array = cv2.GaussianBlur(mask_array, (9, 9), 2.0)
@@ -607,12 +612,13 @@ class ControlNetProcessor:
607
  # Konvertierung von NumPy-Array auf PIL-Image
608
  mask_original = Image.fromarray(mask_array).convert("L")
609
  raw_mask = Image.fromarray(raw_mask_array).convert("L")
 
610
 
611
  # Finale Maske für ControlNet
612
  mask = mask_original
613
 
614
  print(f"✅ FOCUS_CHANGE Maske erstellt: {mask.size}")
615
- return mask, raw_mask
616
 
617
  #Wichtig: mask (SAM-Maske) muß in Originalgröße zurück sonst Probleme in Funktion create_mask_and_sam_together
618
  #Und raw_mask auch in Originalgröße sonst Anzeige-Problem UI
 
594
 
595
  kernel_dilate = np.ones((15, 15), np.uint8)
596
  mask_array = cv2.dilate(mask_array, kernel_dilate, iterations=1)
597
+
598
+ #Sicherung der Binärmaske für Inpaint ohne Graupixel
599
+ inpaint_binary_mask = mask_array.copy()
600
+
601
+ print(f"🔳 [FOCUS] Inpainting-Maske gespeichert: {np.unique(inpaint_binary_mask)}")
602
 
603
  # 3. Weiche Übergänge mittlerer Blur für natürliche Übergänge
604
  mask_array = cv2.GaussianBlur(mask_array, (9, 9), 2.0)
 
612
  # Konvertierung von NumPy-Array auf PIL-Image
613
  mask_original = Image.fromarray(mask_array).convert("L")
614
  raw_mask = Image.fromarray(raw_mask_array).convert("L")
615
+ inpaint_binary_pil = Image.fromarray(inpaint_binary_mask).convert("L") #
616
 
617
  # Finale Maske für ControlNet
618
  mask = mask_original
619
 
620
  print(f"✅ FOCUS_CHANGE Maske erstellt: {mask.size}")
621
+ return mask, raw_mask, inpaint_binary_pil
622
 
623
  #Wichtig: mask (SAM-Maske) muß in Originalgröße zurück sonst Probleme in Funktion create_mask_and_sam_together
624
  #Und raw_mask auch in Originalgröße sonst Anzeige-Problem UI