Astridkraft commited on
Commit
d34a6cb
·
verified ·
1 Parent(s): 2752bcd

Update sam_module.py

Browse files
Files changed (1) hide show
  1. sam_module.py +6 -7
sam_module.py CHANGED
@@ -809,15 +809,14 @@ def create_sam_mask(self, image, bbox_coords, mode):
809
 
810
  # 3. BBOX-ÜBERLAPPUNG (20%)
811
  bbox_mask = np.zeros((original_image.height, original_image.width), dtype=np.uint8)
 
812
  bbox_mask[original_bbox[1]:original_bbox[3], original_bbox[0]:original_bbox[2]] = 1
813
- overlap = np.sum(mask_binary & bbox_mask)
814
-
815
- mask_binary = (mask_np > adaptive_threshold).astype(np.uint8) # Originalgröße!
816
- mask_area_pixels = np.sum(mask_binary) # Zählt korrekt die weißen Pixel
817
  bbox_overlap_ratio = overlap / mask_area_pixels if mask_area_pixels > 0 else 0
818
-
819
- print(f" 🔍 Maske {i+1}: Fläche={mask_area_pixels:,}px, "
820
- f"Überlappung={overlap:,}px ({bbox_overlap_ratio:.1%})")
821
 
822
 
823
  # Für Kopf: Sollte großteils in BBox sein (mind. 70%)
 
809
 
810
  # 3. BBOX-ÜBERLAPPUNG (20%)
811
  bbox_mask = np.zeros((original_image.height, original_image.width), dtype=np.uint8)
812
+
813
  bbox_mask[original_bbox[1]:original_bbox[3], original_bbox[0]:original_bbox[2]] = 1
814
+
815
+ overlap = np.sum(mask_binary & bbox_mask)
816
+
 
817
  bbox_overlap_ratio = overlap / mask_area_pixels if mask_area_pixels > 0 else 0
818
+
819
+ print(f" 📍 BBox-Überlappung: {overlap:,} von {mask_area_pixels:,} Pixeln ({bbox_overlap_ratio:.1%})")
 
820
 
821
 
822
  # Für Kopf: Sollte großteils in BBox sein (mind. 70%)