Update sam_module.py

sam_module.py

@@ -547,25 +547,81 @@ def create_sam_mask(self, image, bbox_coords, mode):
             crop_y1 = max(0, crop_y1)
             crop_x2 = min(original_image.width, crop_x2)
             crop_y2 = min(original_image.height, crop_y2)
+
+
+            # ITERATIVE ANPASSUNG für bessere Crop-Größe
+max_iterations = 3
+print(f"   🔄 Iterative Crop-Anpassung (max. {max_iterations} Versuche)")
+
+for iteration in range(max_iterations):
+    actual_crop_width = crop_x2 - crop_x1
+    actual_crop_height = crop_y2 - crop_y1
     
-            # Falls Crop zu klein ist, anpassen
-            actual_crop_width = crop_x2 - crop_x1
-            actual_crop_height = crop_y2 - crop_y1
+    # Prüfen ob Crop groß genug ist
+    if actual_crop_width >= crop_size and actual_crop_height >= crop_size:
+        print(f"   ✅ Crop-Größe OK nach {iteration} Iteration(en): {actual_crop_width}×{actual_crop_height} px")
+        break
     
-            if actual_crop_width < crop_size or actual_crop_height < crop_size:
-                # An Kanten anpassen
-                if crop_x1 == 0:
-                    crop_x2 = min(original_image.width, crop_size)
-                elif crop_x2 == original_image.width:
-                    crop_x1 = max(0, original_image.width - crop_size)
-        
-                if crop_y1 == 0:
-                    crop_y2 = min(original_image.height, crop_size)
-                elif crop_y2 == original_image.height:
-                    crop_y1 = max(0, original_image.height - crop_size)
+    print(f"   🔄 Iteration {iteration+1}: Crop zu klein ({actual_crop_width}×{actual_crop_height})")
+    
+    # BREITE anpassen (falls nötig)
+    if actual_crop_width < crop_size:
+        if crop_x1 == 0:  # Am linken Rand
+            crop_x2 = min(original_image.width, crop_x1 + crop_size)
+            print(f"     ← Breite angepasst (linker Rand): crop_x2 = {crop_x2}")
+        elif crop_x2 == original_image.width:  # Am rechten Rand
+            crop_x1 = max(0, crop_x2 - crop_size)
+            print(f"     → Breite angepasst (rechter Rand): crop_x1 = {crop_x1}")
+        else:
+            # Nicht am Rand - zentriert erweitern
+            missing_width = crop_size - actual_crop_width
+            expand_left = missing_width // 2
+            expand_right = missing_width - expand_left
+            
+            crop_x1 = max(0, crop_x1 - expand_left)
+            crop_x2 = min(original_image.width, crop_x2 + expand_right)
+            print(f"     ↔ Zentriert erweitert um {missing_width}px")
     
-            print(f"   🔲 Crop-Bereich: [{crop_x1}, {crop_y1}, {crop_x2}, {crop_y2}]")
-            print(f"   📏 Tatsächliche Crop-Größe: {crop_x2-crop_x1} × {crop_y2-crop_y1} px")
+    # HÖHE anpassen (falls nötig)
+    if actual_crop_height < crop_size:
+        if crop_y1 == 0:  # Am oberen Rand
+            crop_y2 = min(original_image.height, crop_y1 + crop_size)
+            print(f"     ↑ Höhe angepasst (oberer Rand): crop_y2 = {crop_y2}")
+        elif crop_y2 == original_image.height:  # Am unteren Rand
+            crop_y1 = max(0, crop_y2 - crop_size)
+            print(f"     ↓ Höhe angepasst (unterer Rand): crop_y1 = {crop_y1}")
+        else:
+            # Nicht am Rand - zentriert erweitern
+            missing_height = crop_size - actual_crop_height
+            expand_top = missing_height // 2
+            expand_bottom = missing_height - expand_top
+            
+            crop_y1 = max(0, crop_y1 - expand_top)
+            crop_y2 = min(original_image.height, crop_y2 + expand_bottom)
+            print(f"     ↕ Zentriert erweitert um {missing_height}px")
+    
+    # Sicherstellen, dass innerhalb der Bildgrenzen
+    crop_x1 = max(0, crop_x1)
+    crop_y1 = max(0, crop_y1)
+    crop_x2 = min(original_image.width, crop_x2)
+    crop_y2 = min(original_image.height, crop_y2)
+    
+    # Letzte Iteration erreicht?
+    if iteration == max_iterations - 1:
+        actual_crop_width = crop_x2 - crop_x1
+        actual_crop_height = crop_y2 - crop_y1
+        print(f"   ⚠️  Max. Iterationen erreicht. Finaler Crop: {actual_crop_width}×{actual_crop_height} px")
+        
+        # Warnung wenn immer noch zu klein
+        if actual_crop_width < crop_size or actual_crop_height < crop_size:
+            min_acceptable = int(bbox_max_dim * 1.8)  # Mindestens 1.8× BBox
+            if actual_crop_width < min_acceptable or actual_crop_height < min_acceptable:
+                print(f"   🚨 KRITISCH: Crop immer noch zu klein ({actual_crop_width}×{actual_crop_height})")
+                print(f"   🚨 SAM könnte Probleme haben!")
+
+            print(f"   🔲 Finaler Crop-Bereich: [{crop_x1}, {crop_y1}, {crop_x2}, {crop_y2}]")
+            print(f"   📏 Finale Crop-Größe: {crop_x2-crop_x1} × {crop_y2-crop_y1} px")
+
     
             # Bild ausschneiden- 2,5 mal so groß und quadratisch wie BBox
             cropped_image = original_image.crop((crop_x1, crop_y1, crop_x2, crop_y2))