Spaces:
Running
Running
Orkhan Hasanli commited on
Commit ·
cd66f30
1
Parent(s): bb55c4d
Expanding smaller objects inside person/car crop by 60%
Browse files- 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 |
-
|
| 396 |
-
|
|
|
|
|
|
|
|
|
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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))
|