Update controlnet_module.py

controlnet_module.py

@@ -450,6 +450,7 @@ class ControlNetProcessor:
 
                 for i in range(num_masks):
                     single_mask = outputs.pred_masks[:, :, i, :, :]
+                    # Interpolation auf Originalgröße
                     resized_mask = F.interpolate(
                         single_mask,
                         size=(image.height, image.width),
@@ -529,16 +530,9 @@ class ControlNetProcessor:
                 
                 print(f"✅ Beste Maske: Nr. {best_mask_idx+1} mit Score {best_score:.3f}")
                 
-                # NUR DIE BESTE MASKE AUF 512x512 HERUNTERSKALIEREN -Für Inpaint
-                best_mask_256 = outputs.pred_masks[:, :, best_mask_idx, :, :]
-                resized_mask = F.interpolate(
-                    best_mask_256,
-                    size=(512, 512),  # DIREKT AUF CONTROLNET-ZIELGRÖßE
-                    mode='bilinear',
-                    align_corners=False
-                ).squeeze()
+                best_mask_original = all_masks[best_mask_idx]        
                 
-                mask_np = resized_mask.cpu().numpy()
+                mask_np = best_mask_original
                 print(f"   🔄 Beste Maske skaliert auf 512×512 für ControlNet")
                 
                 # ============================================================
@@ -608,11 +602,11 @@ class ControlNetProcessor:
                 mask_array = (mask_array_float * 255).astype(np.uint8)
                 
                 # 5. Auf Originalgröße für Rückgabe (falls benötigt)
-                mask_512 = Image.fromarray(mask_array).convert("L")
+                mask_original = Image.fromarray(mask_array).convert("L")
                 raw_mask = Image.fromarray(raw_mask_array).convert("L")
                 
                 # Finale Maske für ControlNet ist 512x512
-                mask = mask_512
+                mask = mask_original
                 
                 print(f"✅ FOCUS_CHANGE Maske erstellt: {mask.size}")
                 return mask, raw_mask     #Wichtig: mask (SAM-Maske) muß in Originalgröße zurück sonst Probleme in Funktion create_mask_and_sam_together