Update controlnet_module.py
Browse files- controlnet_module.py +1 -2
controlnet_module.py
CHANGED
|
@@ -1010,7 +1010,7 @@ class ControlNetProcessor:
|
|
| 1010 |
area_score = optimal_max / area_ratio # Starke Bestrafung für zu groß
|
| 1011 |
|
| 1012 |
|
| 1013 |
-
# 2. BBOX-ÜBERLAPPUNG
|
| 1014 |
bbox_mask = np.zeros((image.height, image.width), dtype=np.uint8)
|
| 1015 |
bbox_mask[bbox_y1:bbox_y2, bbox_x1:bbox_x2] = 1
|
| 1016 |
overlap = np.sum(mask_binary & bbox_mask)
|
|
@@ -1051,7 +1051,6 @@ class ControlNetProcessor:
|
|
| 1051 |
|
| 1052 |
|
| 1053 |
confidence_score = mask_max
|
| 1054 |
-
bbox_score = bbox_overlap_ratio
|
| 1055 |
|
| 1056 |
score = (
|
| 1057 |
bbox_overlap_ratio * 0.5 + # 50% BBox-Überlappung (vorher 20%)
|
|
|
|
| 1010 |
area_score = optimal_max / area_ratio # Starke Bestrafung für zu groß
|
| 1011 |
|
| 1012 |
|
| 1013 |
+
# 2. BBOX-ÜBERLAPPUNG (wie gut überlappt die SAM-Maske die BBox)
|
| 1014 |
bbox_mask = np.zeros((image.height, image.width), dtype=np.uint8)
|
| 1015 |
bbox_mask[bbox_y1:bbox_y2, bbox_x1:bbox_x2] = 1
|
| 1016 |
overlap = np.sum(mask_binary & bbox_mask)
|
|
|
|
| 1051 |
|
| 1052 |
|
| 1053 |
confidence_score = mask_max
|
|
|
|
| 1054 |
|
| 1055 |
score = (
|
| 1056 |
bbox_overlap_ratio * 0.5 + # 50% BBox-Überlappung (vorher 20%)
|