Update controlnet_module.py
Browse files- controlnet_module.py +5 -1
controlnet_module.py
CHANGED
|
@@ -516,7 +516,11 @@ class ControlNetProcessor:
|
|
| 516 |
mask_np = all_masks[best_mask_idx]
|
| 517 |
|
| 518 |
# ============================================================
|
| 519 |
-
# OPTIMIERTER THRESHOLD
|
|
|
|
|
|
|
|
|
|
|
|
|
| 520 |
# ============================================================
|
| 521 |
max_val = mask_np.max()
|
| 522 |
print(f" 🔍 Maximaler SAM-Konfidenzwert der besten Maske: {max_val:.3f}")
|
|
|
|
| 516 |
mask_np = all_masks[best_mask_idx]
|
| 517 |
|
| 518 |
# ============================================================
|
| 519 |
+
# OPTIMIERTER THRESHOLD
|
| 520 |
+
# SAM gibt nur Wahrscheinlichkeiten aus!
|
| 521 |
+
# Nachdem das Modell eine Maske für eine Person vorhersagt (wo jeder Pixel einen Wert zwischen 0 und 1 hat,
|
| 522 |
+
# wie "wahrscheinlich gehört dieser Pixel zur Person"), wird diese Maske binarisiert (0 oder 1), indem alle
|
| 523 |
+
# Pixel unter 0.05 auf 0 gesetzt werden, alle darüber auf 1.
|
| 524 |
# ============================================================
|
| 525 |
max_val = mask_np.max()
|
| 526 |
print(f" 🔍 Maximaler SAM-Konfidenzwert der besten Maske: {max_val:.3f}")
|