Skip to main content

decode_yolo_det

Function decode_yolo_det 

Source
pub fn decode_yolo_det<BOX: PrimInt + AsPrimitive<f32> + Send + Sync>(
    output: (ArrayView2<'_, BOX>, Quantization),
    score_threshold: f32,
    iou_threshold: f32,
    nms: Option<Nms>,
    output_boxes: &mut Vec<DetectBox>,
)
where f32: AsPrimitive<BOX>,
Expand description

Decodes YOLO detection outputs from quantized tensors into detection boxes.

Boxes are expected to be in XYWH format.

Expected shapes of inputs:

  • output: (4 + num_classes, num_boxes)