Update controlnet_module.py
Browse files- 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 |
-
|
| 1087 |
-
compactness_score * 0.3 +
|
| 1088 |
-
|
| 1089 |
-
confidence_score * 0.
|
| 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}:")
|