Megvii-BaseDetection / BorderDet

BorderDet: Border Feature for Dense Object Detection(ECCV2020 Oral)
Apache License 2.0
430 stars 63 forks source link

How do you realize border feature extracting in Table 1? #48

Closed rainylt closed 3 years ago

rainylt commented 3 years ago

In the paper Table 1, you used border features in the second prediction. I'm curious about how you realize it? It's easy to understand that ROI Align is used to extract region features as Mask rcnn, but how do you use the border features to predict the box? Do you fill-up the border-box with zero, or reshape it to a vector and use FC layer to predict the result?

Maycbj commented 3 years ago

In fact, we use RoIAlign to extract border features to predict the box. As shown in the figure, the RoI size is (5, 5). It is very easy to select the index to get the border feature. Then we use the feature to predict the result by FC layers.

image

rainylt commented 3 years ago

In fact, we use RoIAlign to extract border features to predict the box. As shown in the figure, the RoI size is (5, 5). It is very easy to select the index to get the border feature. Then we use the feature to predict the result by FC layers.

image

I got it, thanks for your kind answer!