Yolo
输入
x:
- 是否必填: 必填
- 数据类型: float16,float32
- 参数解释: [batch, boxes *（coords+1+classes）,height,width]
- 规格限制: Float16
属性
- boxes:
- 是否必填: 非必填
- 数据类型: int
- 参数解释: 每个grid的anbox的数量，默认值3
- 规格限制: 无
- coords:
- 是否必填: 非必填
- 数据类型: int
- 参数解释: coords的数量，固定为4，表示x、y、h、w
- 规格限制: 4
- classes:
- 是否必填: 非必填
- 数据类型: int
- 参数解释: 类别数，默认为80，不能超过1024
- 规格限制: 最大1024
- yolo_version:
- 是否必填: 非必填
- 数据类型: string
- 参数解释: “V2”或者“V3”，表示yolo V2和V3的版本；分别对应于detection_layer和Yolo_layer，默认值"V3"
- 规格限制: V2或者V3
- softmax:
- 是否必填: 非必填
- 数据类型: bool
- 参数解释: 在YOLO V2和V3中，表示是否做softmax操作，默认值False
- 规格限制: True或者False
- softmaxtree:
- 是否必填: 非必填
- 数据类型: bool
- 参数解释: softmaxtree固定为False，接口保留（darknet中有定义）
- 规格限制: False
- background:
- 是否必填: 非必填
- 数据类型: bool
- 参数解释: 在YOLO V2和V3中，与“softmax”参数配合，用于确定对b、classes等做的操作类型，默认值False
- 规格限制: True或者False
输出
- coord_data:
- 是否必填: 必填
- 数据类型: float16,float32
- 参数解释: [batch, boxes*coords, ceil(height*width*2+32, 32)/2]，表示预测的coords。
- 规格限制: Float16
- obj_data:
- 是否必填: 必填
- 数据类型: float16,float32
- 参数解释: [batch, ceil(boxes*height*width *2+32, 32)/2]，此处每个anchor的obj的数值只有1个
- 规格限制: Float16
- classes_data:
- 是否必填: 必填
- 数据类型: float16,float32
- 参数解释: [bacth, classes, ceil(boxes*height*width *2+32, 32)/2]，为便于内部计算已将每个anchor的score向16取整
- 规格限制: Float16
父主题： 支持Caffe算子清单