Frcnn里ROIhead的参数解答

问题遇到的现象和发生背景

frcnn参数解读

问题相关代码,请勿粘贴截图

① self._fast_rcnn_loc_loss(roi_loc, gt_roi_loc, gt_roi_label.data, self.roi_sigma)
② roi_cls_locs = self.cls_loc(fc7)

我想要达到的结果

① 搞清楚roi_loc和gt_roi_loc这两个参数分别是什么框对什么框的偏移量 区别是什么?

在最后的RoIHead里②
这个偏移量是建议框对什么框的偏移量呢?

我刚好写过关于边界框回归的博文,你可以看看https://blog.csdn.net/weixin_43694096/article/details/121457923

想明白了 自问自答:
rpn网络和rcnn网络的获取偏移量loss的基准不同
rpn网络的基准是anchor(先验框),两个框偏移量参数分别是anchor对建议框的偏移量、anchor对GT框的偏移量
rcnn网络的基准是rio(建议框),两个框偏移量参数分别是建议框对最终预测框的偏移量、建议框对GT框的偏移量