Astridkraft commited on
Commit
e558370
·
verified ·
1 Parent(s): a668efe

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +15 -4
controlnet_module.py CHANGED
@@ -534,13 +534,24 @@ class ControlNetProcessor:
534
  print(f" ✅ SAM ist sicher für Gesicht (max_val={max_val:.3f} >= 0.8)")
535
 
536
  print(f" 🎯 Gesichts-Threshold: {dynamic_threshold:.3f}")
537
- else:
538
- # Standard Threshold
 
 
 
 
 
 
 
 
 
 
 
 
539
  if max_val < 0.6:
540
  dynamic_threshold = 0.3
541
- print(f" ⚠️ SAM ist unsicher (max_val={max_val:.3f} < 0.6)")
542
  else:
543
- dynamic_threshold = max_val * 0.8
544
  print(f" ✅ SAM ist sicher (max_val={max_val:.3f} >= 0.6)")
545
 
546
  print(f" 🎯 Standard-Threshold: {dynamic_threshold:.3f}")
 
534
  print(f" ✅ SAM ist sicher für Gesicht (max_val={max_val:.3f} >= 0.8)")
535
 
536
  print(f" 🎯 Gesichts-Threshold: {dynamic_threshold:.3f}")
537
+
538
+ elif mode == "focus_change":
539
+ # SPEZIALBEHANDLUNG für Fokus-Änderung
540
+ print(" 🎯 FOCUS-CHANGE: Passe Threshold für vollständige Körpermaske an")
541
+ if best_score < 0.7: # Wenn Maskenqualität schlecht ist
542
+ dynamic_threshold = 0.25 # NIEDRIGER Threshold, um mehr Pixel zu behalten
543
+ print(f" ⚠️ Masken-Score niedrig ({best_score:.3f}). Setze Threshold auf {dynamic_threshold:.3f} für maximale Abdeckung.")
544
+ else:
545
+ # Bei guter Maske: moderaten Threshold verwenden
546
+ dynamic_threshold = max(0.4, max_val * 0.5) # Viel niedriger als 0.8!
547
+ print(f" ✅ Gute Maske. Verwende moderaten Threshold: {dynamic_threshold:.3f}")
548
+
549
+ else: # environment_change oder andere
550
+ # Alte Standardlogik (kann beibehalten werden)
551
  if max_val < 0.6:
552
  dynamic_threshold = 0.3
 
553
  else:
554
+ dynamic_threshold = max_val * 0.8
555
  print(f" ✅ SAM ist sicher (max_val={max_val:.3f} >= 0.6)")
556
 
557
  print(f" 🎯 Standard-Threshold: {dynamic_threshold:.3f}")