Update controlnet_module.py

controlnet_module.py

@@ -832,10 +832,33 @@ class ControlNetProcessor:
                     
                     # Keine Bildverbesserung für 512x512 - Original verwenden
                     image = original_image
-                    
-                    # Für SAM verwenden wir die Original-BBox (keine Transformation)
-                    sam_bbox = (x1, y1, x2, y2)
-                    print(f"   ✅ SAM verwendet Original-BBox-Koordinaten")
+
+                ################################
+                #BBox-Berechnung für Heuristik immer Oroginal-BBox
+                ##############################
+
+                if use_crop_strategy:
+                    # Für Crop: Original-BBox auf Crop-Größe skalieren
+                    scale_x = image.width / original_image.width
+                    scale_y = image.height / original_image.height
+                    bbox_x1 = int(original_bbox[0] * scale_x)
+                    bbox_y1 = int(original_bbox[1] * scale_y)
+                    bbox_x2 = int(original_bbox[2] * scale_x)
+                    bbox_y2 = int(original_bbox[3] * scale_y)
+                else:
+                    # Für Original: Direkt Original-BBox verwenden
+                    bbox_x1, bbox_y1, bbox_x2, bbox_y2 = original_bbox
+    
+                # Heuristik-Parameter berechnen (konsistent für beide Fälle)
+                bbox_width = bbox_x2 - bbox_x1
+                bbox_height = bbox_y2 - bbox_y1
+                bbox_area = bbox_width * bbox_height
+                bbox_center = ((bbox_x1 + bbox_x2) // 2, (bbox_y1 + bbox_y2) // 2)
+    
+                print(f"📏 HEURISTIK-BBOX: [{bbox_x1}, {bbox_y1}, {bbox_x2}, {bbox_y2}]")
+                print(f"📐 HEURISTIK-BBOX-Größe: {bbox_width}×{bbox_height}px = {bbox_area:,}px²")
+                print(f"📍 HEURISTIK-BBOX-Zentrum: {bbox_center}")                  
+                print(f"   ✅ SAM verwendet Original-BBox-Koordinaten")
                
 
                 # ============================================================
@@ -844,11 +867,7 @@ class ControlNetProcessor:
                 # ============================================================
 
                 print("-" * 60)
-                print(f"📦 BOUNDING BOX DETAILS FÜR SAM:")
-                print(f"   Bild-Größe für SAM: {image.size}")
-                print(f"   BBox Koordinaten: [{x1}, {y1}, {x2}, {y2}]")
-                print(f"   BBox Dimensionen: {x2-x1}px × {y2-y1}px")
-                
+                print(f"📦 BOUNDING BOX DETAILS FÜR SAM:")               
                 # Vorbereitung für SAM2 - WICHTIG: NUR EINE BBOX
                 print("-" * 60)
                 print("🖼️  BILDAUFBEREITUNG FÜR SAM 2")
@@ -946,30 +965,7 @@ class ControlNetProcessor:
                 # ============================================================
                 print("🤔 HEURISTIK BERECHNEN")
 
-                if use_crop_strategy:
-                    # BBox-Information für Heuristik (IN CROP-KOORDINATEN!)
-                    bbox_center = ((x1 + x2) // 2, (y1 + y2) // 2)
-                    bbox_area = (x2 - x1) * (y2 - y1)
-    
-                    print(f"   CROP-BBox: [{x1}, {y1}, {x2}, {y2}]")
-                    print(f"   CROP-BBox-Größe: {x2-x1}×{y2-y1}px = {bbox_area:,}px²")
-                    print(f"   CROP-BBox-Zentrum: {bbox_center}")
-                    print(f"   Crop-Bildgröße: {image.width}×{image.height}")
-
-                else: #Heuristik für Original-Fall, Original-BBox Bild<=512
-                    print("   ✅ Original-Fall: Heuristik mit Original-BBox")
-                    
-                    # Original-BBox-Koordinaten verwenden (nicht transformiert)
-                    bbox_center = ((original_bbox[0] + original_bbox[2]) // 2, 
-                                  (original_bbox[1] + original_bbox[3]) // 2)
-                    bbox_area = (original_bbox[2] - original_bbox[0]) * (original_bbox[3] - original_bbox[1])
-    
-                    print(f"   ORIGINAL-BBox: {original_bbox}")
-                    print(f"   ORIGINAL-BBox-Größe: {original_bbox[2]-original_bbox[0]}×{original_bbox[3]-original_bbox[1]}px = {bbox_area:,}px²")
-                    print(f"   ORIGINAL-BBox-Zentrum: {bbox_center}")
-                    print(f"   Original-Bildgröße: {image.width}×{image.height}")
-                    
-               
+                      
                 # Heuristik-BBox für alle weiteren Berechnungen verwenden
                 # (bbox_center und bbox_area sind jetzt korrekt für beide Fälle)