YoloV2DetectionOutput
输入
- coord_data:
- 是否必填: 必填
 - 数据类型: float16、float32
 - 参数解释: [batch,coords *boxes,height,width],表示预测的coords
 - 规格限制: float16,height*width*Dtype_Size>=32 Byte
 
 - obj_prob:
- 是否必填: 必填
 - 数据类型: float16、float32
 - 参数解释: [batch,boxes,height,width],此处每个anchor的obj的数值只有1个
 - 规格限制: float16
 
 - classes_prob:
- 是否必填: 必填
 - 数据类型: float16、float32
 - 参数解释: [batch,boxes*classes,height,width],为了方便AICORE计算已将每个anchor的score向16取整
 - 规格限制: float16,height*width*Dtype_Size>=32 Byte
 
 - img_info:
- 是否必填: 必填
 - 数据类型: float16、float32
 - 参数解释: 原图信息,[batch,4],4表示netH、netW、scaleH、scaleW四个维度;其中netH、netW为网络模型输入的HW,scaleH、scaleW为原始图片的HW
 - 规格限制: float16
 
 
属性
- biases:
- 是否必填: 必填
 - 数据类型: ListFloat
 - 参数解释: [boxes,2],其中2分别表示x(w)和y(h)方向
 - 规格限制: 无
 
 
- boxes:
- 是否必填: 非必填
 - 数据类型: int
 - 参数解释: 每个grid的anbox的数量,默认值5
 - 规格限制: 无
 
 
- coords:
- 是否必填: 非必填
 - 数据类型: int
 - 参数解释: coords的数量,固定为4,表示x、y、h、w
 - 规格限制: 固定为4
 
 
- classes:
- 是否必填: 非必填
 - 数据类型: int
 - 参数解释: 类别数,默认为20
 - 规格限制: 最大1024
 
 
- relative:
- 是否必填: 非必填
 - 数据类型: bool
 - 参数解释: 在correct_region_boxes中表示是否为相对值,True
 - 规格限制: True或者False
 
 
- obj_threshold:
- 是否必填: 非必填
 - 数据类型: float
 - 参数解释: 有物体概率的阈值,对应于clsProb中的阈值,默认值为0.5
 - 规格限制: [0,1]
 
 
- pre_nms_topn:
- 是否必填: 非必填
 - 数据类型: int
 - 参数解释: pre_nms_topn对应于multiClassNMS(对于每个类别,取前pre_nms_topn个数量进行处理,SoC最大支持512;Mini/Cloud最大支持1024,默认值512)
 - 规格限制: 最大1024
 
 
- post_nms_topn:
- 是否必填: 非必填
 - 数据类型: int
 - 参数解释: 经过nms之后返回全部还是postTopK个框,最大为1024;默认值512
 - 规格限制: 最大1024
 
 
- score_threshold:
- 是否必填: 非必填
 - 数据类型: float
 - 参数解释: 每个类别的阈值,默认值为0.5
 - 规格限制: [0,1]
 
 
- iou_threshold:
- 是否必填: 非必填
 - 数据类型: float
 - 参数解释: 交并比(Intersection over Union)阈值,默认值是0.45
 - 规格限制: [0,1]
 
 
输出
- box_out:
- 是否必填: 必填
 - 数据类型: float16、float32
 - 参数解释: [batch,6*post_nms_topn],其中的6表示x1、y1、x2、y2、score、label(类别),实际按box_out_num的数量输出
 - 规格限制: float16
 
 - box_out_num:
 
父主题: 支持Caffe算子清单