Astridkraft commited on
Commit
a1d70f8
·
verified ·
1 Parent(s): 0902eb1

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +5 -2
controlnet_module.py CHANGED
@@ -142,7 +142,7 @@ class ControlNetProcessor:
142
  print("🌳 MODUS: ENVIRONMENT_CHANGE")
143
  print("-" * 60)
144
 
145
- working_mask = mask_array.copy() # Person = Weiß, Hintergrund = Schwarz
146
  print(f"working_mask shape: {working_mask.shape}")
147
 
148
  # Der Prozessor von SAM erwartet ein NumPy-Array kein PIL
@@ -312,9 +312,12 @@ class ControlNetProcessor:
312
  mask_array = np.array(temp_mask) #np. heißt mache aus PIL-Image wieder numPy-Array
313
  print(f" ✅ Maske auf Originalgröße skaliert: {mask_array.shape}")
314
 
 
 
 
315
 
316
  # DILATE auf der weißen Person - daduch wird Person etwas vergrößert
317
- kernel_dilate = np.ones((5, 5), np.uint8)
318
  working_mask = cv2.dilate(working_mask, kernel_dilate, iterations=1)
319
  print(f" ✅ Dilate (5x5) - Person leicht erweitert")
320
 
 
142
  print("🌳 MODUS: ENVIRONMENT_CHANGE")
143
  print("-" * 60)
144
 
145
+
146
  print(f"working_mask shape: {working_mask.shape}")
147
 
148
  # Der Prozessor von SAM erwartet ein NumPy-Array kein PIL
 
312
  mask_array = np.array(temp_mask) #np. heißt mache aus PIL-Image wieder numPy-Array
313
  print(f" ✅ Maske auf Originalgröße skaliert: {mask_array.shape}")
314
 
315
+
316
+ working_mask = mask_array.copy() # Person = Weiß, Hintergrund = Schwarz
317
+
318
 
319
  # DILATE auf der weißen Person - daduch wird Person etwas vergrößert
320
+ kernel_dilate = np.ones((5, 5), np.uint8)
321
  working_mask = cv2.dilate(working_mask, kernel_dilate, iterations=1)
322
  print(f" ✅ Dilate (5x5) - Person leicht erweitert")
323