下载
中文
注册

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]

    • 规格限制: 无