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算子清单