Documentation

RPNProposalSSD

Input

  • rpn_cls_prob_reshape:
    • Required: yes
    • Type: float16
    • Description: foreground/background score, feature_map_size = 32, C = num_anchors * 2
    • Restriction: NCHW, with shape [1, 30, 32, 32]
  • rpn_bbox_pred:
    • Required: yes
    • Type: float16
    • Description:

      BBox coordinates

      C = num_anchors * 4, feature_map_size = 32

    • Restriction: NCHW, with shape [1, 60, 32, 32]
  • im_info:
    • Required: yes
    • Type: float16
    • Description: image information
    • Restriction: NCHW, with shape [1, 6, 1, 1]

Attribute

  • anchor_height:
    • Required: yes
    • Type: listFloat
    • Description: anchor height
    • Restrictions:
  • anchor_width:
    • Required: yes
    • Type: listFloat
    • Description: anchor width
    • Restrictions:
  • bbox_mean:
    • Required: yes
    • Type: listFloat
    • Description: mean value for BBox normalization
    • Restriction: either 0 or 4
  • bbox_std:
    • Required: yes
    • Type: listFloat
    • Description: standard deviation for BBox normalization
    • Restriction: either 0 or 4
  • intype:
    • Required: yes
    • Type: str
    • Description: input data type
    • Restriction: Must be float16.
  • top_n:
    • Required: yes
    • Type: listInt
    • Description: number of output ROIs
    • Restriction: Must be 300.
  • min_size_mode:
    • Required: yes
    • Type: str
    • Description: minimum size mode
    • Restriction: none
  • min_size_h:
    • Required: yes
    • Type: float
    • Description: minimum BBox height
    • Restriction: Must be greater than 0.
  • min_size_w:
    • Required: yes
    • Type: float
    • Description: minimum BBox width
    • Restriction: Must be greater than 0.
  • heat_map_a:
    • Required: yes
    • Type: listFloat
    • Description: scaling ratio
    • Restriction: Must be greater than 0.
  • overlap_ratio:
    • Required: yes
    • Type: listFloat
    • Description: IoU threshold of NMS
    • Restriction: Must be in the range [0, 1].
  • threshold_objectness:
    • Required: yes
    • Type: float
    • Description: foreground threshold
    • Restriction: Must be in the range [0, 1].
  • max_candidate_n:
    • Required: yes
    • Type: listFloat
    • Description: maximum number of candidate BBoxes for NMS
    • Restriction: Must be 3000.
  • refine_out_of_map_bbox:
    • Required: yes
    • Type: bool
    • Description: whether to refine out-of-map BBoxes
    • Restriction: default = true
  • use_soft_nms:
    • Required: yes
    • Type: listBool
    • Description: whether to apply soft NMS
    • Restriction: default = false
  • voting:
    • Required: yes
    • Type: listBool
    • Description: default = false
    • Restriction: none
  • vote_iou:
    • Required: yes
    • Type: listFloat
    • Description: default = 0.7
    • Restriction: none

Output

rois:
  • Required: yes
  • Type: float16
  • Description: output ROIs
  • Restriction: NCHW, with shape [n, top_n, 8, 1]