Spaces:
Running
Running
Update sam_module.py
Browse files- 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 |
-
|
| 814 |
-
|
| 815 |
-
|
| 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 |
-
|
| 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%)
|