Documentation

Proposal

Input

  • cls_prob:
    • Required: yes
    • Type: float16, float32
    • Description:

      Foreground and background probabilities formatted as [batch,num_anchors*2,height,width]. *2 indicates the background (bg prob) and foreground (fg probs).

    • Restrictions: none
  • bbox_delta:
    • Required: yes
    • Type: float16, float32
    • Description: preselected box position formatted as (batch, num_anchors*4, height, width)
    • Restrictions: none
  • im_info:
    • Required: no
    • Type: -
    • Description: height and width of the input image (default = {375, 1240})
    • Restrictions: none

Attribute

  • feat_stride:
    • Required: yes
    • Type: float
    • Description: stride for the sliding blocks (default = 16)
    • Restrictions: none
  • base_size:
    • Required: no
    • Type: float
    • Description: base size of the anchor in generate_anchors (default = 16)
    • Restrictions: none
  • min_size:
    • Required: no
    • Type: float
    • Description: minimum size of box (default = {16, 16})
    • Restrictions: none
  • ratio:
    • Required: no
    • Type: ListFloat
    • Description: anchor ratios of height to width in generate_anchors (default = [0.5, 1, 2])
    • Restrictions: none
  • scale:
    • Required: no
    • Type: ListFloat
    • Description: anchor sizes relative to base_size in generate_anchors (default = {8, 16, 32})
    • Restrictions: none
  • pre_nms_topn:
    • Required: no
    • Type: int
    • Description: K in topk. With float16 input, up to 3008 for Ascend AI Processor; else, up to 6008. With float32 input, up to 1502 (default) for Ascend AI Processor; else, up to 3004.
    • Restriction: Must be within 1502.
  • post_nms_topn:
    • Required: no
    • Type: int
    • Description: number of final outputs. Up to 3000 for Ascend AI Processor; else, up to 6000. With float32 input, up to 1502 for Ascend AI Processor; else, up to 3000; defaulted to 304.
    • Restriction: Must be within 1502.
  • iou_threshold:
    • Required: no
    • Type: float
    • Description: Intersection over Union (IoU) threshold (default = 0.7). Must be in the range (0, 1).
    • Restriction: Must be in the range (0, 1).
  • output_actual_rois_num:
    • Required: no
    • Type: bool
    • Description: If True, adds the actual_rois_num output; if False (default), otherwise.
    • Restrictions: none

Output

  • rois:
    • Required: yes
    • Type: float16, float32
    • Description: a tensor for the output BBox information, with shape [batch, 5-tuple, N], where N is post_nms_topn rounded up to the nearest multiple of 16, the 5-tuple is formatted as (batchID, x1, y1, x2, y2), and the number of BBoxes output per batch is determined by actual_rois_num.
    • Restrictions: none
  • actual_rois_num:
    • Required: no
    • Type: int32
    • Description: actual number of BBoxes output per batch. Has shape [batch, 8]. The value 8 indicates that only the first column in the eight columns is valid (data padding is performed for performance acceleration, and the rest seven columns are invalid data that is padded).

      For example, for shape [4, 8], only the first column is valid, indicating the number of BBoxes output per batch.

      [0,0]...[0,7]

      [1,0]...[1,7]

      [2,0]...[2,7]

      [3,0]...[3,7]

    • Restrictions: none