Faster-rcnn代码中bbox_inside_weights的作用是什么

这篇博文主要综合了知乎上两位答主的答案,此处仅作学习记录作用:

参考连接:https://www.zhihu.com/question/65587875

Faster R-CNN代码中有两个变量比较难以理解:bbox_inside_weights和bbox_outside_weights。在参考了一些回答之后,稍微理解了些,在这里简单记录。

1. bbox_inside_weights

用来设置正样本回归loss的权重,默认为1。

可以看到后半部分的边界框回归,其中Pi *是指只针对前景(fg)来做回归,bbox_inside_weights指的就是这个。

代码块中,一开始将inside_weights初始为全零矩阵,然后在label=1的地方写入(1.0, 1.0, 1.0, 1.0)

2. bbox_outside_weights

而bbox_outside_weights,应该是Nreg,起到一个分母的作用。

猜你喜欢

转载自www.cnblogs.com/elitphil/p/12753309.html