Orkhan Hasanli commited on
Commit
cd66f30
·
1 Parent(s): bb55c4d

Expanding smaller objects inside person/car crop by 60%

Browse files
Files changed (1) hide show
  1. dfine_jina_pipeline.py +10 -3
dfine_jina_pipeline.py CHANGED
@@ -392,8 +392,11 @@ def main():
392
  if obj_w <= 0 or obj_h <= 0:
393
  continue
394
 
395
- pad_x = obj_w * 0.3
396
- pad_y = obj_h * 0.3
 
 
 
397
  bx1 = max(0, int(bx1 - pad_x))
398
  by1 = max(0, int(by1 - pad_y))
399
  bx2 = min(img_w, int(bx2 + pad_x))
@@ -602,7 +605,11 @@ def run_single_image(
602
  if obj_w <= 0 or obj_h <= 0:
603
  continue
604
 
605
- pad_x, pad_y = obj_w * 0.3, obj_h * 0.3
 
 
 
 
606
  bx1 = max(0, int(bx1 - pad_x))
607
  by1 = max(0, int(by1 - pad_y))
608
  bx2 = min(img_w, int(bx2 + pad_x))
 
392
  if obj_w <= 0 or obj_h <= 0:
393
  continue
394
 
395
+ # Small objects (min side < 24 px): expand by 60%; larger: 30%
396
+ min_side_obj = min(obj_w, obj_h)
397
+ pad_ratio = 0.6 if min_side_obj < 24 else 0.3
398
+ pad_x = obj_w * pad_ratio
399
+ pad_y = obj_h * pad_ratio
400
  bx1 = max(0, int(bx1 - pad_x))
401
  by1 = max(0, int(by1 - pad_y))
402
  bx2 = min(img_w, int(bx2 + pad_x))
 
605
  if obj_w <= 0 or obj_h <= 0:
606
  continue
607
 
608
+ # Small objects (min side < 24 px): expand by 60%; larger: 30%
609
+ min_side_obj = min(obj_w, obj_h)
610
+ pad_ratio = 0.6 if min_side_obj < 24 else 0.3
611
+ pad_x = obj_w * pad_ratio
612
+ pad_y = obj_h * pad_ratio
613
  bx1 = max(0, int(bx1 - pad_x))
614
  by1 = max(0, int(by1 - pad_y))
615
  bx2 = min(img_w, int(bx2 + pad_x))