Skip to main content

decode_modelpack_split_quant

Function decode_modelpack_split_quant 

Source
pub fn decode_modelpack_split_quant<D: AsPrimitive<f32>>(
    outputs: &[ArrayView3<'_, D>],
    configs: &[ModelPackDetectionConfig],
    score_threshold: f32,
    iou_threshold: f32,
    output_boxes: &mut Vec<DetectBox>,
)
Expand description

Decodes ModelPack split detection outputs from quantized tensors. The boxes are expected to be in XYWH format.

The configs must correspond to the outputs in order.

Expected shapes of inputs:

  • outputs: (width, height, num_anchors * (5 + num_classes))

§Panics

Panics if shapes don’t match the expected dimensions.