Update sam_module.py
Browse files- sam_module.py +2 -11
sam_module.py
CHANGED
|
@@ -352,17 +352,8 @@ def create_sam_mask(self, image, bbox_coords, mode):
|
|
| 352 |
# Alle 3 Masken analysieren
|
| 353 |
for i in range(num_masks):
|
| 354 |
# Maske in Original-SAM-Größe (256x256) analysieren
|
| 355 |
-
|
| 356 |
-
|
| 357 |
-
|
| 358 |
-
# Für Heuristik: Temporär auf Bildgröße skalieren für Flächenverhältnis und Schwerpunktposition
|
| 359 |
-
temp_mask = F.interpolate(
|
| 360 |
-
mask_256,
|
| 361 |
-
size=(image.height, image.width),
|
| 362 |
-
mode='bilinear',
|
| 363 |
-
align_corners=False
|
| 364 |
-
).squeeze()
|
| 365 |
-
mask_np_temp = temp_mask.sigmoid().cpu().numpy()
|
| 366 |
|
| 367 |
# Adaptive Vor-Filterung (prüft ob Maske überhaupt gültig ist)
|
| 368 |
mask_max = mask_np_temp.max()
|
|
|
|
| 352 |
# Alle 3 Masken analysieren
|
| 353 |
for i in range(num_masks):
|
| 354 |
# Maske in Original-SAM-Größe (256x256) analysieren
|
| 355 |
+
|
| 356 |
+
mask_np_temp = all_masks[i]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 357 |
|
| 358 |
# Adaptive Vor-Filterung (prüft ob Maske überhaupt gültig ist)
|
| 359 |
mask_max = mask_np_temp.max()
|