文档

PSROIPooling

Input

  • x:
    • Required: yes
    • Type: float16
    • Description: feature map of shape [batch, C, H, W], where C = rois_num x group_size^2
    • Restrictions: none
  • rois:
    • Required: yes
    • Type: float16
    • Description: Has shape [batch, 5-tuple, rois_num], where the 5-tuple is formatted as (batchID, x1, y1, x2, y2).
    • Restrictions: none

Attribute

  • spatial_scale:
    • Required: yes
    • Type: float
    • Description: multiplicative spatial scale factor to translate ROI coordinates from their input scale to the scale used when pooling.
    • Restriction: Must be greater than 0.
  • output_dim:
    • Required: yes
    • Type: int
    • Description: number of channels in the pooled output. Must be greater than 0.
    • Restriction: Must be within the channels of the pooled ROI output.
  • group_size:
    • Required: yes
    • Type: int
    • Description: height and width of the output feature map
    • Restriction: The output ROI can have the same H and W sizes. input_channel == out_channel * group_size^2

Output

y:
  • Required: yes
  • Type: float16
  • Description: output of shape [batch * rois_num, output_dim, group_size, group_size] according to the defined channels.
  • Restrictions: none