Astridkraft commited on
Commit
53f7739
·
verified ·
1 Parent(s): 6c51732

Update sam_module.py

Browse files
Files changed (1) hide show
  1. 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 in Original-SAM-Größe (256x256) analysieren
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()