Astridkraft commited on
Commit
a8a1956
·
verified ·
1 Parent(s): 5a62fa0

Update sam_module.py

Browse files
Files changed (1) hide show
  1. sam_module.py +2 -2
sam_module.py CHANGED
@@ -65,7 +65,7 @@ def create_sam_mask(self, image, bbox_coords, mode):
65
  num_masks = outputs.pred_masks.shape[2]
66
  print(f" SAM lieferte {num_masks} verschiedene Masken")
67
 
68
- # ✅ WICHTIG: Sammlung aller Masken hinzufügen (fehlt in Ihrem Code!)
69
  all_masks = []
70
 
71
  for i in range(num_masks):
@@ -335,7 +335,7 @@ def create_sam_mask(self, image, bbox_coords, mode):
335
 
336
  num_masks = outputs.pred_masks.shape[2]
337
 
338
- # ✅ WICHTIG: Sammlung aller Masken hinzufügen (fehlt in Ihrem Code!)
339
  all_masks = []
340
 
341
  for i in range(num_masks):
 
65
  num_masks = outputs.pred_masks.shape[2]
66
  print(f" SAM lieferte {num_masks} verschiedene Masken")
67
 
68
+ # Sammlung aller Masken in all_masks
69
  all_masks = []
70
 
71
  for i in range(num_masks):
 
335
 
336
  num_masks = outputs.pred_masks.shape[2]
337
 
338
+ # Sammlung aller Masken in all_masks
339
  all_masks = []
340
 
341
  for i in range(num_masks):