Astridkraft commited on
Commit
23e99aa
·
verified ·
1 Parent(s): 699799a

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. 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}")