下载
中文
注册

PSROIPooling

输入

  • x:
    • 是否必填: 必填
    • 数据类型: float16
    • 参数解释: feature Map,[batch,C,H,W], 其中的C=rois_num*group_size^2
    • 规格限制: 无
  • rois:
    • 是否必填: 必填
    • 数据类型: float16
    • 参数解释: Shape是 [batch,5,rois_num],其中的5表示(batchID,x1,y1,x2,y2)
    • 规格限制: 无

属性

  • spatial_scale:
    • 是否必填: 必填
    • 数据类型: float
    • 参数解释: 进行ROIPooling特征输入的尺寸相比原始输入的比例
    • 规格限制: 大于0
  • output_dim:
    • 是否必填: 必填
    • 数据类型: int
    • 参数解释: 输出channels,必须大于0
    • 规格限制: 支持输出ROI的channel维度大小
  • group_size:
    • 是否必填: 必填
    • 数据类型: int
    • 参数解释: 输出Feature的高和宽
    • 规格限制: 支持输出ROI的H和W大小,H=W;要求原始的input_channel= =out_channel*group_size^2

输出

y:
  • 是否必填: 必填
  • 数据类型: float16
  • 参数解释: 类似ROIPooling,不同位置的输出来自不同channel的输入[batch*rois_num,output_dim,group_size,group_size]
  • 规格限制: 无