Update controlnet_module.py
Browse files- controlnet_module.py +3 -2
controlnet_module.py
CHANGED
|
@@ -603,7 +603,7 @@ class ControlNetProcessor:
|
|
| 603 |
mask_array_float = mask_array_float ** 0.85
|
| 604 |
mask_array = (mask_array_float * 255).astype(np.uint8)
|
| 605 |
|
| 606 |
-
#
|
| 607 |
mask_original = Image.fromarray(mask_array).convert("L")
|
| 608 |
raw_mask = Image.fromarray(raw_mask_array).convert("L")
|
| 609 |
|
|
@@ -611,7 +611,8 @@ class ControlNetProcessor:
|
|
| 611 |
mask = mask_original
|
| 612 |
|
| 613 |
print(f"✅ FOCUS_CHANGE Maske erstellt: {mask.size}")
|
| 614 |
-
return mask, raw_mask
|
|
|
|
| 615 |
#Wichtig: mask (SAM-Maske) muß in Originalgröße zurück sonst Probleme in Funktion create_mask_and_sam_together
|
| 616 |
#Und raw_mask auch in Originalgröße sonst Anzeige-Problem UI
|
| 617 |
|
|
|
|
| 603 |
mask_array_float = mask_array_float ** 0.85
|
| 604 |
mask_array = (mask_array_float * 255).astype(np.uint8)
|
| 605 |
|
| 606 |
+
# Konvertierung von NumPy-Array auf PIL-Image
|
| 607 |
mask_original = Image.fromarray(mask_array).convert("L")
|
| 608 |
raw_mask = Image.fromarray(raw_mask_array).convert("L")
|
| 609 |
|
|
|
|
| 611 |
mask = mask_original
|
| 612 |
|
| 613 |
print(f"✅ FOCUS_CHANGE Maske erstellt: {mask.size}")
|
| 614 |
+
return mask, raw_mask
|
| 615 |
+
|
| 616 |
#Wichtig: mask (SAM-Maske) muß in Originalgröße zurück sonst Probleme in Funktion create_mask_and_sam_together
|
| 617 |
#Und raw_mask auch in Originalgröße sonst Anzeige-Problem UI
|
| 618 |
|