文档

YoloNms

Input

x1:
  • Required: yes
  • Type: float16
  • Description: output of Yolov5FourInputDecodeBox
  • Restriction: NCHW

Attribute

  • shape:
    • Required: yes
    • Type: int
    • Description: the sum of h*w of the four input feature_maps of the Yolov5FourInputDecodeBox operator
  • thresh:
    • Required: yes
    • Type: float16
    • Description: NMS threshold
    • Restriction: Must be in the range [0.0, 1.0].
  • num_anchor:
    • Required: yes
    • Type: int
    • Description: the number of anchors
    • Restriction: Must be greater than 0.
  • num_class:
    • Required: yes
    • Type: int
    • Description: number of classes
    • Restriction: Must be in the range [1, 80].
  • total_output_proposal_num:
    • Required: yes
    • Type: int
    • Description: the number of boxes to be output per class
    • Restriction: Must be in the range [5, 50].

Output

y:
  • Required: yes
  • Type: float16
  • Description: output of SSMH network postprocessing after NMS, [total_output_proposal_num*class_num, 8]
  • Restriction: NCHW. The output boxes are arranged by type. [total_output_proposal_num*class_num, :4] indicates the coordinate, [total_output_proposal_num*class_num, 4] indicates the score, and [total_output_proposal_num*class_num, 5:] is a reserved bit.