Update controlnet_module.py
Browse files- controlnet_module.py +17 -3
controlnet_module.py
CHANGED
|
@@ -1298,9 +1298,23 @@ class ControlNetProcessor:
|
|
| 1298 |
print("⚠️ Kein Pose-Detector verfügbar, verwende Fallback")
|
| 1299 |
return self.extract_pose_simple(image)
|
| 1300 |
|
| 1301 |
-
print(" Extrahiere Pose mit OpenPose
|
| 1302 |
-
|
| 1303 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1304 |
return pose_image
|
| 1305 |
except Exception as e:
|
| 1306 |
print(f"Fehler bei Pose-Extraktion: {e}")
|
|
|
|
| 1298 |
print("⚠️ Kein Pose-Detector verfügbar, verwende Fallback")
|
| 1299 |
return self.extract_pose_simple(image)
|
| 1300 |
|
| 1301 |
+
print(" Extrahiere Pose mit OpenPose und allen Gelenkpunkten")
|
| 1302 |
+
|
| 1303 |
+
pose_image = detector(
|
| 1304 |
+
image,
|
| 1305 |
+
include_body=True,
|
| 1306 |
+
include_hand=True, # 🔥 Hände einschließen (21 Punkte pro Hand)
|
| 1307 |
+
include_face=True, # 🔥 Gesicht einschließen (70 Punkte)
|
| 1308 |
+
hand_and_face=True, # 🔥 Beide gleichzeitig
|
| 1309 |
+
detect_resolution=768, # 🔥 Höhere Detektionsauflösung für Details
|
| 1310 |
+
image_resolution=512, # Ausgabegröße
|
| 1311 |
+
return_pil=True
|
| 1312 |
+
)
|
| 1313 |
+
print("✅ Detaillierte Pose-Map erstellt")
|
| 1314 |
+
print(f" 🔥 137 Gelenkpunkte (statt nur 25)")
|
| 1315 |
+
print(f" 🔥 Enthält: Körper (25) + Hände (42) + Gesicht (70)")
|
| 1316 |
+
print(f" 🔥 Detektionsauflösung: 768px für mehr Details")
|
| 1317 |
+
|
| 1318 |
return pose_image
|
| 1319 |
except Exception as e:
|
| 1320 |
print(f"Fehler bei Pose-Extraktion: {e}")
|