Update sam_module.py
Browse files- sam_module.py +2 -14
sam_module.py
CHANGED
|
@@ -90,20 +90,8 @@ def create_sam_mask(self, image, bbox_coords, mode):
|
|
| 90 |
best_score = -1
|
| 91 |
|
| 92 |
# Alle 3 Masken analysieren (OHNE sie alle zu skalieren!)
|
| 93 |
-
for i in range(num_masks):
|
| 94 |
-
# Maske
|
| 95 |
-
mask_256 = outputs.pred_masks[:, :, i, :, :]
|
| 96 |
-
mask_np_256 = mask_256.sigmoid().squeeze().cpu().numpy()
|
| 97 |
-
|
| 98 |
-
# Für Heuristik: Temporär auf Bildgröße skalieren für Flächenverhältnis und Schwerpunktposition
|
| 99 |
-
temp_mask = F.interpolate(
|
| 100 |
-
mask_256,
|
| 101 |
-
size=(image.height, image.width),
|
| 102 |
-
mode='bilinear',
|
| 103 |
-
align_corners=False
|
| 104 |
-
).squeeze()
|
| 105 |
-
|
| 106 |
-
mask_np_temp = temp_mask.sigmoid().cpu().numpy()
|
| 107 |
|
| 108 |
# Adaptive Vor-Filterung (prüft ob Maske überhaupt gültig ist)
|
| 109 |
mask_max = mask_np_temp.max()
|
|
|
|
| 90 |
best_score = -1
|
| 91 |
|
| 92 |
# Alle 3 Masken analysieren (OHNE sie alle zu skalieren!)
|
| 93 |
+
for i in range(num_masks):
|
| 94 |
+
mask_np_temp = all_masks[i] #verwende Maske auf Original-Bildgröße
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 95 |
|
| 96 |
# Adaptive Vor-Filterung (prüft ob Maske überhaupt gültig ist)
|
| 97 |
mask_max = mask_np_temp.max()
|