Update controlnet_module.py
Browse files- 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 |
|