Update sam_module.py
Browse files- 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 |
-
#
|
| 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 |
-
#
|
| 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):
|