Update controlnet_module.py
Browse files- controlnet_module.py +24 -26
controlnet_module.py
CHANGED
|
@@ -253,34 +253,32 @@ class ControlNetProcessor:
|
|
| 253 |
print(f"Fehler beim Laden von Multi-ControlNet Outside: {e}")
|
| 254 |
raise
|
| 255 |
return self.pipe_multi_outside
|
| 256 |
-
|
| 257 |
-
|
| 258 |
-
def prepare_controlnet_maps(self, image, keep_environment=False):
|
| 259 |
-
"""
|
| 260 |
-
ERSTELLT NUR CONDITIONING-MAPS, generiert KEIN Bild.
|
| 261 |
-
"""
|
| 262 |
-
print("🎯 ControlNet: Erstelle Conditioning-Maps...")
|
| 263 |
-
|
| 264 |
-
if keep_environment:
|
| 265 |
-
# Depth + Canny
|
| 266 |
-
print(" Modus: Depth + Canny")
|
| 267 |
-
conditioning_images = [
|
| 268 |
-
self.extract_depth_map(image),
|
| 269 |
-
self.extract_canny_edges(image)
|
| 270 |
-
]
|
| 271 |
-
else:
|
| 272 |
-
# OpenPose + Canny
|
| 273 |
-
print(" Modus: OpenPose + Canny")
|
| 274 |
-
conditioning_images = [
|
| 275 |
-
self.extract_pose(image),
|
| 276 |
-
self.extract_canny_edges(image)
|
| 277 |
-
]
|
| 278 |
-
|
| 279 |
-
print(f"✅ {len(conditioning_images)} Conditioning-Maps erstellt.")
|
| 280 |
-
return conditioning_images # Rückgabe: Liste der PIL Images
|
| 281 |
|
| 282 |
|
| 283 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 284 |
|
| 285 |
def prepare_inpaint_input(self, image, keep_environment=False):
|
| 286 |
"""
|
|
|
|
| 253 |
print(f"Fehler beim Laden von Multi-ControlNet Outside: {e}")
|
| 254 |
raise
|
| 255 |
return self.pipe_multi_outside
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 256 |
|
| 257 |
|
| 258 |
+
def prepare_controlnet_maps(self, image, keep_environment=False):
|
| 259 |
+
"""
|
| 260 |
+
ERSTELLT NUR CONDITIONING-MAPS, generiert KEIN Bild.
|
| 261 |
+
"""
|
| 262 |
+
print("🎯 ControlNet: Erstelle Conditioning-Maps...")
|
| 263 |
+
|
| 264 |
+
if keep_environment:
|
| 265 |
+
# Depth + Canny
|
| 266 |
+
print(" Modus: Depth + Canny")
|
| 267 |
+
conditioning_images = [
|
| 268 |
+
self.extract_depth_map(image),
|
| 269 |
+
self.extract_canny_edges(image)
|
| 270 |
+
]
|
| 271 |
+
else:
|
| 272 |
+
# OpenPose + Canny
|
| 273 |
+
print(" Modus: OpenPose + Canny")
|
| 274 |
+
conditioning_images = [
|
| 275 |
+
self.extract_pose(image),
|
| 276 |
+
self.extract_canny_edges(image)
|
| 277 |
+
]
|
| 278 |
+
|
| 279 |
+
print(f"✅ {len(conditioning_images)} Conditioning-Maps erstellt.")
|
| 280 |
+
return conditioning_images # Rückgabe: Liste der PIL Images
|
| 281 |
+
|
| 282 |
|
| 283 |
def prepare_inpaint_input(self, image, keep_environment=False):
|
| 284 |
"""
|