Update sam_module.py

sam_module.py

@@ -62,11 +62,26 @@ def create_sam_mask(self, image, bbox_coords, mode):
                 print("   Führe Vorhersage durch...")
                 outputs = self.sam_model(**inputs)
                 print(f"✅ Vorhersage abgeschlossen")
-                print(f"   Anzahl der Vorhersagemasken: {outputs.pred_masks.shape[2]}")
+                print(f"   Anzahl der Vorhersagemasken: {outputs.pred_masks.shape[2]}")               
 
             num_masks = outputs.pred_masks.shape[2]
             print(f"   SAM lieferte {num_masks} verschiedene Masken")  
 
+            #Sammlung der Masken
+            all_masks = []
+            
+            for i in range(num_masks):
+                single_mask = outputs.pred_masks[:, :, i, :, :]
+                resized_mask = F.interpolate(
+                    single_mask,
+                    size=(image.height, image.width),
+                    mode='bilinear',
+                    align_corners=False
+                ).squeeze()
+                
+                mask_np = resized_mask.sigmoid().cpu().numpy()
+                all_masks.append(mask_np)
+
             bbox_center = ((x1 + x2) // 2, (y1 + y2) // 2)
             bbox_area = (x2 - x1) * (y2 - y1)
             print(f"   Erwartetes BBox-Zentrum: {bbox_center}")
@@ -535,4 +550,5 @@ def create_sam_mask(self, image, bbox_coords, mode):
             print(f"   ⚠️  Fallback-Maske angepasst: {fallback_mask.size} → {original_image.size}")
             fallback_mask = fallback_mask.resize(original_image.size, Image.Resampling.NEAREST)
         
-        return fallback_mask, fallback_mask
+        return fallback_mask, fallback_mask
+