Astridkraft commited on
Commit
2c7d8e9
·
verified ·
1 Parent(s): 519b893

Update controlnet_module.py

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