文档

Yolo3dNMS

Input

bbox:
  • Required: yes
  • Type: float16
  • Description: input bounding box
  • Restriction: NCHW; supported dimension: [1,32,576,90]
scores:
  • Required: yes
  • Type: float16
  • Description: confidence of the input bounding box
  • Restriction: NCHW; supported dimension: [1,9,576,90]
cls_cnt:
  • Required: yes
  • Type: int32
  • Description: number of classes
  • Restriction: NCHW; supported dimension: [1,16,1,1]
brvis_cnt:
  • Required: yes
  • Type: int32
  • Description: number of visible brake lights
  • Restriction: NCHW; supported dimension: [1,16,1,1]
brswt_cnt:
  • Required: yes
  • Type: int32
  • Description: number of brake lights that are on
  • Restriction: NCHW; supported dimension: [1,16,1,1]
ltvis_cnt:
  • Required: yes
  • Type: int32
  • Description: number of visible left turn lights
  • Restriction: NCHW; supported dimension: [1,16,1,1]
ltswt_cnt:
  • Required: yes
  • Type: int32
  • Description: number of left turn lights that are on
  • Restriction: NCHW; supported dimension: [1,16,1,1]
rtvis_cnt:
  • Required: yes
  • Type: int32
  • Description: number of visible right turn lights
  • Restriction: NCHW; supported dimension: [1,16,1,1]
rtswt_cnt:
  • Required: yes
  • Type: int32
  • Description: number of right turn lights that are on
  • Restriction: NCHW; supported dimension: [1,16,1,1]
proposal_gm:
  • Required: yes
  • Type: float16
  • Description: memory used for auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,51840,8]
proposal_index:
  • Required: yes
  • Type: float16
  • Description: index of auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,51840,8]
topk_gm:
  • Required: yes
  • Type: float16
  • Description: topk memory used for auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,51904,8]
first_sorted_index:
  • Required: yes
  • Type: float16
  • Description: index of auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,8192,8]
second_sorted_index:
  • Required: yes
  • Type: float16
  • Description: index of auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,304,8]
first_box_gm:
  • Required: yes
  • Type: float16
  • Description: memory used for auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,32,316]
first_score_gm:
  • Required: yes
  • Type: float16
  • Description: memory used for auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,8,316]
first_proposal:
  • Required: yes
  • Type: float16
  • Description: memory used for auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,304,8]
first_index:
  • Required: yes
  • Type: float16
  • Description: memory used for auxiliary calculation
  • Restriction: NCHW; supported dimension: [1,304,8]

Attribute

first_stage_size:
  • Required: yes
  • Type: int64
  • Description: number of bounding boxes in the first phase
second_stage_size:
  • Required: yes
  • Type: int64
  • Description: number of bounding boxes in the second phase
cls_nms_thresh:
  • Required: yes
  • Type: float32
  • Description: class threshold
nms_thresh:
  • Required: yes
  • Type: float32
  • Description: NMS threshold
conf_thresh:
  • Required: yes
  • Type: float32
  • Description: class confidence

Output

output_nms:
  • Required: yes
  • Type: float16
  • Description: result of the output bounding box
  • Restriction: NCHW; supported dimension: [1,40,116,1]