psroi_pool

    注意 rois必须为2维LoDTensor,lod_level为1

    该OP执行PSROIPooling运算,是位置敏感的感兴趣区域池化方法(Position sensitive region of interest pooling,也称为PSROIPooling)。输入input是位置敏感的评分图,输入rois是感兴趣区域的位置坐标。PSROIPooling不同于普通ROIPooling的地方在于,输入input特征图的不同通道会跟输出特征图上的位置区域相关联,该方法是在R-FCN模型中首次提出来的,更多详细信息请参阅 。

    样例

    参数:

    • rois (Variable) - 感兴趣区域,2-D LoDTensor,形状为(num_rois, 4),lod_level为1。形式如[x1, y1, x2, y2], …],其中(x1, y1)为左上角坐标,(x2, y2)为右下角坐标。数据类型与input相同

    • output_channels (int) - 输出特征图的通道数。对于共C个种类的图像分类任务,output_channels应该是 (C + 1) ,其中1代表背景

    • pooled_height (int) - 池化输出的高度

    • name (str|None) - 该参数供开发人员打印调试信息时使用,具体用法请参见 Name ,默认值为None。

    返回: 经过PSROIPooling之后的结果,形状为(num_rois,output_channels,pooled_height,pooled_width) 的4维LoDTensor,lod_level为1,数据类型与input相同,与rois具有相同的lod信息。

    返回类型: Variable

    代码示例:

    1. x = fluid.layers.data(name='x', shape=[490, 28, 28], dtype='float32')
    2. pool_out = fluid.layers.psroi_pool(x, rois, 10, 1.0, 7, 7)