Update controlnet_module.py
Browse files- 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
|