Faster RCNN网络源码解读（Ⅵ） --- RPN网络代码解析（上）RPNHead类与AnchorsGenerator类解析

目录

一、代码作用（rpn_function.py）

二、代码解析 

2.1 RPNHead类

2.2 AnchorsGenerator类

2.2.1 初始化函数__init__

2.2.2 正向传播过程 forward

2.2.3 set_cell_anchors生成anchors模板 

2.2.4 generate_anchors生成anchors 

2.2.5 cached_grid_anchors 

2.2.6 grid_anchors 

---

一、代码作用（rpn_function.py）

          如何利用backbone生成的预测特征层来进行目标分数、目标边界框边界的预测以及如何利用AnchorGeneraror生成anchor。

二、代码解析 

2.1 RPNHead类

class RPNHead(nn.Module):
    """
    add a RPN head with classification and regression
    通过滑动窗口计算预测目标概率与bbox regression参数

    Arguments:
        in_channels