Astridkraft commited on
Commit
8d8dd67
·
verified ·
1 Parent(s): ec9b93f

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. 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
- pose_image = detector(image, hand_and_face=True)
1303
- print("✅ Pose-Map erfolgreich erstellt")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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}")