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.