Astridkraft commited on
Commit
bfba606
·
verified ·
1 Parent(s): 2960512

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. 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
- #Doppelte Berechnung: CROP und Original damit Heuristik
869
- # auf Original berechnet werden kann und Weiterverarbeitung auf Crop
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, :, :]