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)