Mrinal Jain
commited on
Increase `ar_thr` from 20 to 100 for better detection on slender (high aspect ratio) objects (#5556)
Browse files* Making `ar_thr` available as a hyperparameter
* Disabling ar_thr as hyperparameter and computing from the dataset instead
* Fixing bug in ar_thr computation
* Fix `ar_thr` to 100
- utils/augmentations.py +1 -1
utils/augmentations.py
CHANGED
@@ -269,7 +269,7 @@ def mixup(im, labels, im2, labels2):
|
|
269 |
return im, labels
|
270 |
|
271 |
|
272 |
-
def box_candidates(box1, box2, wh_thr=2, ar_thr=
|
273 |
# Compute candidate boxes: box1 before augment, box2 after augment, wh_thr (pixels), aspect_ratio_thr, area_ratio
|
274 |
w1, h1 = box1[2] - box1[0], box1[3] - box1[1]
|
275 |
w2, h2 = box2[2] - box2[0], box2[3] - box2[1]
|
|
|
269 |
return im, labels
|
270 |
|
271 |
|
272 |
+
def box_candidates(box1, box2, wh_thr=2, ar_thr=100, area_thr=0.1, eps=1e-16): # box1(4,n), box2(4,n)
|
273 |
# Compute candidate boxes: box1 before augment, box2 after augment, wh_thr (pixels), aspect_ratio_thr, area_ratio
|
274 |
w1, h1 = box1[2] - box1[0], box1[3] - box1[1]
|
275 |
w2, h2 = box2[2] - box2[0], box2[3] - box2[1]
|