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) – 偏移系数，控制偏移量的大小。