Update controlnet_module.py
Browse files- controlnet_module.py +3 -5
controlnet_module.py
CHANGED
|
@@ -864,14 +864,12 @@ class ControlNetProcessor:
|
|
| 864 |
num_masks = outputs.pred_masks.shape[2]
|
| 865 |
print(f" SAM lieferte {num_masks} verschiedene Masken")
|
| 866 |
|
| 867 |
-
|
| 868 |
-
#
|
| 869 |
-
|
| 870 |
-
#==============
|
| 871 |
|
| 872 |
# Masken speichern in den Arrays
|
| 873 |
all_masks_crop = [] #Weiterverarbeitung in Crop-Größe
|
| 874 |
-
all_masks_original = [] #Heuristikberechnung besser in Originalgröße!
|
| 875 |
|
| 876 |
for i in range(num_masks):
|
| 877 |
single_mask = outputs.pred_masks[:, :, i, :, :]
|
|
|
|
| 864 |
num_masks = outputs.pred_masks.shape[2]
|
| 865 |
print(f" SAM lieferte {num_masks} verschiedene Masken")
|
| 866 |
|
| 867 |
+
#========================
|
| 868 |
+
# Heuristik und Postprocessing auf Crop
|
| 869 |
+
#=========================
|
|
|
|
| 870 |
|
| 871 |
# Masken speichern in den Arrays
|
| 872 |
all_masks_crop = [] #Weiterverarbeitung in Crop-Größe
|
|
|
|
| 873 |
|
| 874 |
for i in range(num_masks):
|
| 875 |
single_mask = outputs.pred_masks[:, :, i, :, :]
|