Update controlnet_module.py
Browse files- controlnet_module.py +1 -1
controlnet_module.py
CHANGED
|
@@ -225,7 +225,7 @@ class ControlNetProcessor:
|
|
| 225 |
mask_np = final_mask.sigmoid().cpu().numpy()
|
| 226 |
print(f" Nach Sigmoid und CPU: {mask_np.shape}, Wertebereich: [{mask_np.min():.3f}, {mask_np.max():.3f}]")
|
| 227 |
|
| 228 |
-
mask_array = (mask_np > 0.
|
| 229 |
unique_vals = np.unique(mask_array)
|
| 230 |
print(f" Nach Threshold (0.5): {mask_array.shape}, Unique Werte: {unique_vals}")
|
| 231 |
|
|
|
|
| 225 |
mask_np = final_mask.sigmoid().cpu().numpy()
|
| 226 |
print(f" Nach Sigmoid und CPU: {mask_np.shape}, Wertebereich: [{mask_np.min():.3f}, {mask_np.max():.3f}]")
|
| 227 |
|
| 228 |
+
mask_array = (mask_np > 0.9).astype(np.uint8) * 255
|
| 229 |
unique_vals = np.unique(mask_array)
|
| 230 |
print(f" Nach Threshold (0.5): {mask_array.shape}, Unique Werte: {unique_vals}")
|
| 231 |
|