roi_align

    实现RoIAlign操作。

    Region of Interests align(直译:有意义、有价值选区对齐) 用于实现双线性插值,它可以将不均匀大小的输入 变为固定大小的特征图(feature map)。

    该运算通过 和 将每个推荐区域划分为等大小分块。位置保持不变。

    在每个RoI分块中,分别取sampling_ratio个点(若为-1则取框内所有点),每个点通过双线性插值直接计算得到坐标。再对分块内取的点取平均值作为小框的坐标值。坐标对齐有误的问题。

    • input (Variable) – 维度为[N,C,H,W]的4-D Tensor,N 为batch大小, C 为输入通道的个数, H 特征高度, W 特征宽度。数据类型为float32或float64。

    • pooled_height (int32,可选) – 池化后的输出高度,默认值为1。

    • name (str,可选) – 具体用法请参见 ,一般无需设置,默认值为None。

    返回:表示RoI align输出的LoDTensor,数据类型为float32或float64,维度为 (num_rois, channels, pooled_h, pooled_w)

    返回类型:Variable