Update controlnet_module.py
Browse files- 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 |
-
|
| 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 |
|