search

MingtaoFu / gliding_vertex

The implementation of paper "Gliding vertex on the horizontal bounding box for multi-oriented object detection".
271 stars 63 forks source link

five extra target variables （TPAMI论文中提到的额外添加的五个输出参数在code中的具体位置） #42

Open Easonsen opened 3 years ago

Easonsen commented 3 years ago

您好！您论文中提到了只是在faster RCNN head上加了5个额外的输出参数，请问在代码中是如何体现这5个额外参数的呢？（很抱歉打扰您了）

Easonsen commented 3 years ago

在gliding_vertex/maskrcnn_benchmark/modeling/roi_heads/box_head/roi_box_predictors.py中

NOTE Mingtao @registry.ROI_BOX_PREDICTOR.register("FPNRatioPredictor") @registry.ROI_BOX_PREDICTOR.register("RatioPredictor") class FPNRatioPredictor(nn.Module):

这个类是不是就是您定义的输出呢？ bbox_pred就是文中的 ‘horizontal bounding box (x, y, w, h)’

而fix_pred就是文中的 'four variables (alpha1, alpha2, alpha3, alpha4) characterizing the oriented bounding box'

而ratio_pred就是文中的 'obliquity factor r that indicates whether the object is nearly horizontal or not'

如有错误，望您指正。

qimw commented 3 years ago

是的， fix 可以看作是 gliding_vertex 的一个alias