2018-04-23-FasterRCNN-rpn_msr

1. proposal_layer.py(产生proposal的文件)

proposal的产生过程:

(1)根据预设的scales和aspect ratios,甚至angles获取anchors

(2)anchors经过bounding box regression的反变换bbox_transform_inv,获取proposals

(3)使用_filter_boxes去除尺寸太小的proposals

(4)按照score大小取若干proposals(Train:12000 Test:6000)

(5)应用nms进行抑制(Train:2000, test:300)

(6)再次取若干proposals并返回第一列batch_inds的rois

2. anchor_target_layer.py(计算rpn网络损失的文件)

(1)根据预设的scales和aspect ratios,甚至angles获取anchors

(2)计算anchor与gt的重叠率

(3)根据overlap对anchor的类别进行设置,并在数量太多的时候进行随机采样,随机采样的样本类别设置为-1

(4)计算anchor与gt的偏差

3. proposal_target_layer.py

(1) 将gt及抖动的gt一起加到rois中

(2)计算rois和gt的重叠率

(3)根据overlap对rois进行选择(正样本的重叠率>0.5, 0<负样本<0.5)

注:

rpn计算损失时,只有部分感兴趣的anchor才会参与到计算过程中

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容