Skip to main content

impl_yolo_quant

edgefirst_decoder::yolo

Function impl_yolo_quant 

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

Internal implementation of YOLO decoding for quantized tensors.

Expected shapes of inputs:

  • output: (4 + num_classes, num_boxes)