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)[source]#

Deformable PSROI(Position Sensitive Region of Interest) Pooling.

Parameters:

  • inp (Tensor) – input feature map.

  • rois (Tensor) – the rois for feature pooling.

  • trans (Tensor) – input offset to psroi_pooling.

  • no_trans (bool) – check the phase of DeformablePSROIPooling. False to the 1st phase, True to the 2nd phase.

  • part_size (int) – part size.

  • sample_per_part (int) – sample points of each part.

  • pooled_shape – kernel shape of convolution.

  • spatial_scale (float) – the spatial_scale w.r.t input image.

  • trans_std (float) – multiplier used in 2nd phase.