megengine.functional.nn.deformable_psroi_pooling

deformable_psroi_pooling(inp, rois, trans, no_trans, part_size, pooled_h, pooled_w, sample_per_part, spatial_scale, trans_std=0.1)[源代码]

可变形的位置敏感的感兴趣区域池化

参数
  • inp (Tensor) – 卷积运算的特征图。

  • rois (Tensor) – 对特征池化的感兴趣区域。

  • trans (Tensor) – PSROI池化的输入坐标偏移。

  • no_trans (bool) – 确定PSROI池化阶段是否加入偏移以获取新的输出。如果为True则表示不加入偏移,进入第二阶段,如果为False就进入第一阶段。

  • part_size (int) – 偏移的高度和宽度。

  • sample_per_part (int) – 每个部分中的样本数量。

  • pooled_shape – 卷积核的形状。

  • spatial_scale (float) – 输入特征图的高度(或宽度)与原始图像高度(或宽度)的比率。

  • trans_std (float) – 偏移系数,控制偏移量的大小。