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]
- 规格限制: 无
父主题: 支持Caffe算子清单