Astridkraft commited on
Commit
2cb2270
·
verified ·
1 Parent(s): c1fbed8

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +10 -2
controlnet_module.py CHANGED
@@ -1176,9 +1176,17 @@ class ControlNetProcessor:
1176
  print(f"👤 Crop={crop_size}×{crop_size}px, Heuristik-Score={best_score:.3f}")
1177
  print(f"👤 Kopfabdeckung: {coverage_ratio:.1%} der BBox")
1178
  print("#" * 80)
1179
-
1180
 
1181
- return mask_512, raw_mask_512 #in app.py wird mask immer auf 512x512 skaliert - hier ist Fehler: Maske und Raw_mask muß in Originalgröße zurück!
 
 
 
 
 
 
 
 
 
1182
 
1183
  # ============================================================
1184
  # UNBEKANNTER MODUS
 
1176
  print(f"👤 Crop={crop_size}×{crop_size}px, Heuristik-Score={best_score:.3f}")
1177
  print(f"👤 Kopfabdeckung: {coverage_ratio:.1%} der BBox")
1178
  print("#" * 80)
 
1179
 
1180
+
1181
+ # Besser: Auf Originalgröße skalieren
1182
+ mask_original = mask_512.resize(original_size, Image.Resampling.LANCZOS)
1183
+ raw_mask_original = raw_mask_512.resize(original_size, Image.Resampling.NEAREST)
1184
+
1185
+ print(f"🔄 Masken auf Originalgröße skaliert: {mask_512.size} → {mask_original.size}")
1186
+
1187
+ return mask_original, raw_mask_original
1188
+
1189
+ #return mask_512, raw_mask_512 #in app.py wird mask immer auf 512x512 skaliert - hier ist Fehler: Maske und Raw_mask muß in Originalgröße zurück!
1190
 
1191
  # ============================================================
1192
  # UNBEKANNTER MODUS