RCNNProposal
输入
- cls_score_softmax:- 是否必填:必填
- 数据类型:float16
- 参数解释:分类置信度
- 规格限制:NCHW,shape为[b*topn1,4,1,1],b表示batch数,topn1表示RPNProposalSSD层的top_n
 
- bbox_pred:- 是否必填:必填
- 数据类型:float16
- 参数解释:预测框的坐标偏移,用于微调rois
- 规格限制:NCHW,属性regress_agnostic选项为“true”的时候shape为[b* topn1,8,1,1](前景和背景),否则为[b* topn1,16,1,1](背景和3种类别),b表示batch数,topn1表示RPNProposalSSD层的top_n
 
- rois:- 是否必填:必填
- 数据类型:float16
- 参数解释:预测框的坐标
- 规格限制:NCHW,shape为[b,topn1,8,1],b表示batch数,topn1表示RPNProposalSSD层的top_n
 
- im_info:- 是否必填:必填
- 数据类型:float16
- 参数解释:图像信息
- 规格限制:NCHW,shape为固定值[b,6,1,1],b表示batch数
 
属性
- bbox_mean:- 是否必填:必填
- 数据类型:listFloat
- 参数解释:bbox归一化均值
- 规格限制:size必须是0或4
 
- bbox_std:- 是否必填:必填
- 数据类型:listFloat
- 参数解释:bbox归一化标准差
- 规格限制:size必须是0或4
 
- num_class:- 是否必填:非必填
- 数据类型:int
- 参数解释:类别数量
- 规格限制:仅支持3
 
- rpn_proposal_output_score:- 是否必填:非必填
- 数据类型:bool
- 参数解释:是否输出各类别置信度
- 规格限制:仅支持true
 
- regress_agnostic:- 是否必填:非必填
- 数据类型:bool
- 参数解释:是否区分前景类别
- 规格限制:无
 
- min_size_h:- 是否必填:必填
- 数据类型:float
- 参数解释:bbox高度最小值,默认值为8.8008
- 规格限制:无
 
- min_size_w:- 是否必填:必填
- 数据类型:float
- 参数解释:bbox宽度最小值,默认值为8.8008
- 规格限制:无
 
- min_size_mode:- 是否必填:必填
- 数据类型:str
- 参数解释:最小值模式,枚举型,默认值为HEIGHT_OR_WIDTH(HEIGHT_OR_WIDTH=0,HEIGHT_OR_WIDTH=1)
- 规格限制:无
 
- threshold_objectness:- 是否必填:必填
- 数据类型:float
- 参数解释:前景阈值
- 规格限制:取值范围(0,1)
 
- threshold:- 是否必填:必填
- 数据类型:listFloat
- 参数解释:各类别置信度阈值,size等于num_class
- 规格限制:取值范围(0,1)
 
- refine_out_of_map_bbox:- 是否必填:必填
- 数据类型:bool
- 参数解释:是否调整bbox到框内
- 规格限制:无
 
- overlap_ratio:- 是否必填:必填
- 数据类型:listFloat
- 参数解释:nms交并比阈值
- 规格限制:无
 
- top_n:- 是否必填:必填
- 数据类型:listInt
- 参数解释:nms输出框的数目
- 规格限制:无
 
- max_candidate_n:- 是否必填:必填
- 数据类型:listInt
- 参数解释:nms候选框最大数目
- 规格限制:无
 
- bsz01:- 是否必填:非必填
- 数据类型:float
- 参数解释:bbox尺寸增加值
- 规格限制:无
 
- allow_border:- 是否必填:非必填
- 数据类型:float
- 参数解释:允许超出图像边界像素数目
- 规格限制:无
 
- allow_border_ratio:- 是否必填:非必填
- 数据类型:float
- 参数解释:允许超出图像边界面积比例
- 规格限制:无
 
输出
bboxes:
- 是否必填:必填
- 数据类型:float16
- 参数解释:输出框的信息,包括该框所在图像编号,框的坐标以及各类别置信度:| img_idx | x1 | y1 | x2 | y2 | background_score | cls1_score | cls2_score | cls3_score |
- 规格限制:NCHW,shape为[b,topn,9,1],b表示batch数,topn表示本算子的top_n
父主题: 支持Caffe算子清单