Proposal

输入

cls_prob:
- 是否必填: 必填
- 数据类型: float16、float32
- 参数解释:
  前景和背景的概率，[batch,num_anchors*2,height,width] ，*2表示背景和前景概率（第一个是背景概率bg prob），第二个是前景概率（fg probs）
- 规格限制: 无

bbox_delta:
- 是否必填: 必填
- 数据类型: float16、float32
- 参数解释: 预选框位置(Preselected box position)，（batch,num_anchors*4,height,Width）
- 规格限制: 无

im_info:
- 是否必填: 非必填
- 数据类型: -
- 参数解释: 输入图片的大小（高和宽），默认值为{375, 1240}
- 规格限制: 无

属性

feat_stride:
- 是否必填: 必填
- 数据类型: float
- 参数解释: 滑块滑动的大小，默认值16
- 规格限制: 无

base_size:
- 是否必填: 非必填
- 数据类型: float
- 参数解释: generate_anchors中生成基本框的大小，默认值16
- 规格限制: 无

min_size:
- 是否必填: 非必填
- 数据类型: float
- 参数解释: box的最小大小，默认值:{16,16}
- 规格限制: 无

ratio:
- 是否必填: 非必填
- 数据类型: ListFloat
- 参数解释: generate_anchors中生成框的高宽比例，默认值为[0.5,1,2]
- 规格限制: 无

scale:
- 是否必填: 非必填
- 数据类型: ListFloat
- 参数解释: generate_anchors中生成框的高宽相对于base_size的大小，默认值为{8,16,32}
- 规格限制: 无

pre_nms_topn:
- 是否必填: 非必填
- 数据类型: int
- 参数解释: Topk中的K，在float16场景下，非昇腾AI处理器场景中最大为6008，昇腾AI处理器场景中最大3008；当采用FP32时，非昇腾AI处理器场景中最大为3004，昇腾AI处理器场景中最大1502（默认值1502）
- 规格限制: 不超过1502

post_nms_topn:
- 是否必填: 非必填
- 数据类型: int
- 参数解释: 最终的输出数量，非昇腾AI处理器场景中最大为6000，昇腾AI处理器场景最大3000；当采用FP32时，非昇腾AI处理器场景中最大为3000，昇腾AI处理器场景中最大1502（默认值304）
- 规格限制: 不超过1502

iou_threshold:
- 是否必填: 非必填
- 数据类型: float
- 参数解释: 交并比（Intersection over Union）的阈值，必须是 (0,1)，默认为(0.7)
- 规格限制: (0,1)

output_actual_rois_num:
- 是否必填: 非必填
- 数据类型: bool
- 参数解释: 是否输出actual_rois_num，False: 不输出，默认为False；True: 输出
- 规格限制: 无

输出

rois:
- 是否必填: 必填
- 数据类型: float16、float32
- 参数解释: 输出框信息，Tensor[batch, 5, N]，其中N是post_nms_topn对齐到16的倍数的值，5表示（batchID ,x1,y1,x2,y2,），实际每个batch按actual_rois_num输出
- 规格限制: 无

actual_rois_num:
- 是否必填: 非必填
- 数据类型: int32
- 参数解释: [batch,8]，其中8表示8列数中只有第一列有效（为满足性能加速，进行数据对齐，后面7列数字是补齐的无效数据），表示实际每个batch输出的bbox数量
   例如，Shape为[4,8]，其中只有首列表示有效数据，存放每个batch的bbox数量
  
  [0,0]...[0,7]
  
  [1,0]...[1,7]
  
  [2,0]...[2,7]
  
  [3,0]...[3,7]
- 规格限制: 无

父主题： 支持Caffe算子清单