Update controlnet_module.py
Browse files- 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 |
-
|
| 538 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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}")
|