Update controlnet_module.py
Browse files- controlnet_module.py +7 -5
controlnet_module.py
CHANGED
|
@@ -832,14 +832,14 @@ class ControlNetProcessor:
|
|
| 832 |
|
| 833 |
# Keine Bildverbesserung für 512x512 - Original verwenden
|
| 834 |
image = original_image
|
| 835 |
-
#x1, y1, x2, y2 bleiben Originalkoordinaten
|
| 836 |
|
| 837 |
-
################################
|
| 838 |
#BBox-Berechnung für Heuristik immer Oroginal-BBox
|
| 839 |
-
##############################
|
| 840 |
|
| 841 |
if use_crop_strategy:
|
| 842 |
-
#
|
| 843 |
scale_x = image.width / original_image.width
|
| 844 |
scale_y = image.height / original_image.height
|
| 845 |
bbox_x1 = int(original_bbox[0] * scale_x)
|
|
@@ -849,7 +849,9 @@ class ControlNetProcessor:
|
|
| 849 |
else:
|
| 850 |
# Für Original: Direkt Original-BBox verwenden
|
| 851 |
bbox_x1, bbox_y1, bbox_x2, bbox_y2 = original_bbox
|
| 852 |
-
|
|
|
|
|
|
|
| 853 |
# Heuristik-Parameter berechnen (konsistent für beide Fälle)
|
| 854 |
bbox_width = bbox_x2 - bbox_x1
|
| 855 |
bbox_height = bbox_y2 - bbox_y1
|
|
|
|
| 832 |
|
| 833 |
# Keine Bildverbesserung für 512x512 - Original verwenden
|
| 834 |
image = original_image
|
| 835 |
+
#x1, y1, x2, y2 bleiben Originalkoordinaten sind zu Beginn des Blockes gespeichert
|
| 836 |
|
| 837 |
+
##########################################################
|
| 838 |
#BBox-Berechnung für Heuristik immer Oroginal-BBox
|
| 839 |
+
##########################################################
|
| 840 |
|
| 841 |
if use_crop_strategy:
|
| 842 |
+
# Berechnung: wo wäre die Original-BBox im verkleinerten Crop-(Kopf am Rand des Bildes)
|
| 843 |
scale_x = image.width / original_image.width
|
| 844 |
scale_y = image.height / original_image.height
|
| 845 |
bbox_x1 = int(original_bbox[0] * scale_x)
|
|
|
|
| 849 |
else:
|
| 850 |
# Für Original: Direkt Original-BBox verwenden
|
| 851 |
bbox_x1, bbox_y1, bbox_x2, bbox_y2 = original_bbox
|
| 852 |
+
|
| 853 |
+
|
| 854 |
+
#########das folgende ist für beide Fälle####################
|
| 855 |
# Heuristik-Parameter berechnen (konsistent für beide Fälle)
|
| 856 |
bbox_width = bbox_x2 - bbox_x1
|
| 857 |
bbox_height = bbox_y2 - bbox_y1
|