Astridkraft commited on
Commit
796c226
·
verified ·
1 Parent(s): ab06849

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. 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
- # 5. Auf Originalgröße für Rückgabe (falls benötigt)
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