Documentation

DFMBPSROIAlign

Input

  • datain_dic:
    • Required: yes
    • Type: float16
    • Description: Input data
    • Restriction: NCHW (H = W = 32)
  • rois_dict:
    • Required: yes
    • Type: float16
    • Description: ROIs
    • Restriction: NCHW

Attribute

  • class_num:
    • Required: no
    • Type: uint32
    • Description: number of classes
    • Restriction: Must be 10.
  • group_height:
    • Required: no
    • Type: uint32
    • Description: group height
    • Restriction: Must be 7.
  • group_width:
    • Required: no
    • Type: uint32
    • Description: group width
    • Restriction: Must be 7.
  • pooled_height:
    • Required: no
    • Type: uint32
    • Description: output height
    • Restriction: Must be 7.
  • pooled_width:
    • Required: no
    • Type: uint32
    • Description: output width
    • Restriction: Must be 7.
  • sample_per_part:
    • Required: no
    • Type: uint32
    • Description: number of samples in each bin
    • Restriction: Must be 4.
  • heat_map_a:
    • Required: no
    • Type: float
    • Description: scaling ratio
    • Restriction: Must be greater than 0.

Output

output_dic:
  • Required: yes
  • Type: float16
  • Description: output tensor
  • Restriction: NCHW