Astridkraft commited on
Commit
2e982b5
·
verified ·
1 Parent(s): 4bb27dd

Update controlnet_module.py

Browse files
Files changed (1) hide show
  1. controlnet_module.py +15 -5
controlnet_module.py CHANGED
@@ -1062,7 +1062,9 @@ class ControlNetProcessor:
1062
 
1063
  overlap = np.sum(mask_binary & bbox_mask)
1064
 
1065
- bbox_overlap_ratio = overlap / mask_area_pixels if mask_area_pixels > 0 else 0
 
 
1066
 
1067
  print(f" 📍 BBox-Überlappung: {overlap:,} von {mask_area_pixels:,} Pixeln ({bbox_overlap_ratio:.1%})")
1068
 
@@ -1082,11 +1084,19 @@ class ControlNetProcessor:
1082
  confidence_score = mask_max
1083
 
1084
  # GESAMTSCORE für Kopf
 
 
 
 
 
 
 
 
1085
  score = (
1086
- area_score * 0.4 + # 40% Flächenpassung
1087
- compactness_score * 0.3 + # 30% Kompaktheit
1088
- bbox_score * 0.2 + # 20% BBox-Überlappung
1089
- confidence_score * 0.1 # 10% Konfidenz
1090
  )
1091
 
1092
  print(f" 📊 GESICHTS-SCORES für Maske {i+1}:")
 
1062
 
1063
  overlap = np.sum(mask_binary & bbox_mask)
1064
 
1065
+ #bbox_overlap_ratio = overlap / mask_area_pixels if mask_area_pixels > 0 else 0
1066
+ bbox_overlap_ratio = overlap / bbox_area if bbox_area > 0 else 0
1067
+
1068
 
1069
  print(f" 📍 BBox-Überlappung: {overlap:,} von {mask_area_pixels:,} Pixeln ({bbox_overlap_ratio:.1%})")
1070
 
 
1084
  confidence_score = mask_max
1085
 
1086
  # GESAMTSCORE für Kopf
1087
+ #score = (
1088
+ # area_score * 0.4 + # 40% Flächenpassung
1089
+ # compactness_score * 0.3 + # 30% Kompaktheit
1090
+ # bbox_score * 0.2 + # 20% BBox-Überlappung
1091
+ # confidence_score * 0.1 # 10% Konfidenz
1092
+ #)
1093
+
1094
+
1095
  score = (
1096
+ bbox_overlap_ratio * 0.5 + # 50% BBox-Überlappung (vorher 20%)
1097
+ compactness_score * 0.3 + # 30% Kompaktheit (vorher 30%)
1098
+ area_score * 0.15 + # 15% Flächenpassung (vorher 40%)
1099
+ confidence_score * 0.05 # 5% Konfidenz (vorher 10%)
1100
  )
1101
 
1102
  print(f" 📊 GESICHTS-SCORES für Maske {i+1}:")