Astridkraft commited on
Commit
9c09208
·
verified ·
1 Parent(s): 59f8710

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +5 -8
controlnet_module.py CHANGED
@@ -658,11 +658,10 @@ class ControlNetProcessor:
658
  print(f" Originalgröße: {original_image.width}×{original_image.height}")
659
 
660
  use_crop_strategy = False
661
-
662
  else:
663
- print(f"📏 Bild ist größer als 512x512 ({original_image.size}) → Verwende Crop-Strategie")
664
-
665
  use_crop_strategy = True
 
666
 
667
  if use_crop_strategy:
668
  #
@@ -924,7 +923,7 @@ class ControlNetProcessor:
924
  # 2. FÜR VERARBEITUNG: Auf CROP-GRÖSSE interpolieren
925
  resized_mask_crop = F.interpolate(
926
  single_mask,
927
- size=(image.height, image.width), # CROP-Größe!
928
  mode='bilinear',
929
  align_corners=False
930
  ).squeeze()
@@ -953,7 +952,7 @@ class ControlNetProcessor:
953
  print(f" CROP-BBox-Zentrum: {bbox_center}")
954
  print(f" Crop-Bildgröße: {image.width}×{image.height}")
955
 
956
- else: #Heuristik für Original-Fall, Original-BBox
957
  print(" ✅ Original-Fall: Heuristik mit Original-BBox")
958
 
959
  # Original-BBox-Koordinaten verwenden (nicht transformiert)
@@ -1367,9 +1366,7 @@ class ControlNetProcessor:
1367
  print(f"👤 Crop={crop_size}×{crop_size}px, Heuristik-Score={best_score:.3f}")
1368
  print(f"👤 Kopfabdeckung: {coverage_ratio:.1%} der BBox")
1369
  print("#" * 80)
1370
-
1371
-
1372
-
1373
 
1374
  return mask_original, raw_mask_original, inpaint_binary_original
1375
 
 
658
  print(f" Originalgröße: {original_image.width}×{original_image.height}")
659
 
660
  use_crop_strategy = False
 
661
  else:
662
+ print(f"📏 Bild ist größer als 512x512 ({original_image.size}) → Verwende Crop-Strategie")
 
663
  use_crop_strategy = True
664
+
665
 
666
  if use_crop_strategy:
667
  #
 
923
  # 2. FÜR VERARBEITUNG: Auf CROP-GRÖSSE interpolieren
924
  resized_mask_crop = F.interpolate(
925
  single_mask,
926
+ size=(image.height, image.width), # CROP-Größe bei Crop-Eingang sonst 512er-Größe
927
  mode='bilinear',
928
  align_corners=False
929
  ).squeeze()
 
952
  print(f" CROP-BBox-Zentrum: {bbox_center}")
953
  print(f" Crop-Bildgröße: {image.width}×{image.height}")
954
 
955
+ else: #Heuristik für Original-Fall, Original-BBox Bild<=512
956
  print(" ✅ Original-Fall: Heuristik mit Original-BBox")
957
 
958
  # Original-BBox-Koordinaten verwenden (nicht transformiert)
 
1366
  print(f"👤 Crop={crop_size}×{crop_size}px, Heuristik-Score={best_score:.3f}")
1367
  print(f"👤 Kopfabdeckung: {coverage_ratio:.1%} der BBox")
1368
  print("#" * 80)
1369
+
 
 
1370
 
1371
  return mask_original, raw_mask_original, inpaint_binary_original
1372