Skip to main content

decode_yolo_det_float

Function decode_yolo_det_float 

Source
pub fn decode_yolo_det_float<T>(
    output: ArrayView2<'_, T>,
    score_threshold: f32,
    iou_threshold: f32,
    nms: Option<Nms>,
    output_boxes: &mut Vec<DetectBox>,
)
where T: Float + AsPrimitive<f32> + Send + Sync + 'static, f32: AsPrimitive<T>,
Expand description

Decodes YOLO detection outputs from float tensors into detection boxes.

Boxes are expected to be in XYWH format.

Expected shapes of inputs:

  • output: (4 + num_classes, num_boxes)