How to get l, r, t, b parameters?

tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)

https://arxiv.org/abs/1904.01355

Other

3.28k stars 630 forks source link

How to get l, r, t, b parameters? #122

Closed SebastienLinker closed 5 years ago

SebastienLinker commented 5 years ago

Hi, I am interested in finding something like the center of gravity of the objects. How do you get those four parameters (fig 1 of the paper) from the network? All I see is the bounding box.

Thanks

tianzhi0549 commented 5 years ago

@ElPatou they are stored in per_box_regression at https://github.com/tianzhi0549/FCOS/blob/efb76e48e6490a93cc8b6b5dc93738fa1df34af5/fcos_core/modeling/rpn/fcos/inference.py#L96.

SebastienLinker commented 5 years ago

Thanks