Astridkraft commited on
Commit
6179a37
·
verified ·
1 Parent(s): 53f7739

Update sam_module.py

Browse files
Files changed (1) hide show
  1. 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
- mask_256 = outputs.pred_masks[:, :, i, :, :]
356
- mask_np_256 = mask_256.sigmoid().squeeze().cpu().numpy()
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()