Expanding by 10% person/car crop to make sure objects centers are still within person/car bbox

dfine_jina_pipeline.py

@@ -108,6 +108,25 @@ def box_center_inside(box, crop_box):
     )
 
 
+def expand_box_by_margin(box, margin_ratio, img_w, img_h):
+    """Expand box [x1,y1,x2,y2] by margin_ratio (e.g. 0.1 = 10%) on all sides, clamped to image."""
+    x1, y1, x2, y2 = box
+    w, h = x2 - x1, y2 - y1
+    if w <= 0 or h <= 0:
+        return box
+    mx = w * margin_ratio
+    my = h * margin_ratio
+    x1 = max(0, x1 - mx)
+    y1 = max(0, y1 - my)
+    x2 = min(img_w, x2 + mx)
+    y2 = min(img_h, y2 + my)
+    return [x1, y1, x2, y2]
+
+
+# 10% margin on person/car group box when testing if object center is inside (avoids missing boundary objects)
+PERSON_CAR_GROUP_MARGIN = 0.10
+
+
 def squarify_crop_box(bx1, by1, bx2, by2, img_w, img_h):
     """
     Expand the shorter side to match the longer (same ratio / square), centered, clamped to image.
@@ -360,10 +379,11 @@ def main():
         for gidx, grp in enumerate(top_groups):
             x1, y1, x2, y2 = grp["box"]
             group_box = [x1, y1, x2, y2]
+            group_box_with_margin = expand_box_by_margin(group_box, PERSON_CAR_GROUP_MARGIN, img_w, img_h)
 
             inside = [
                 d for d in detections
-                if box_center_inside(d["box"], group_box) and d["cls"] not in person_car_ids
+                if box_center_inside(d["box"], group_box_with_margin) and d["cls"] not in person_car_ids
             ]
             inside = deduplicate_by_iou(inside, iou_threshold=0.9)
 
@@ -554,10 +574,11 @@ def run_single_image(
     for gidx, grp in enumerate(top_groups):
         x1, y1, x2, y2 = grp["box"]
         group_box = [x1, y1, x2, y2]
+        group_box_with_margin = expand_box_by_margin(group_box, PERSON_CAR_GROUP_MARGIN, img_w, img_h)
 
         inside = [
             d for d in detections
-            if box_center_inside(d["box"], group_box) and d["cls"] not in person_car_ids
+            if box_center_inside(d["box"], group_box_with_margin) and d["cls"] not in person_car_ids
         ]
         inside = deduplicate_by_iou(inside, iou_threshold=0.9)