Skip to main content

decode_modelpack_split_float

Function decode_modelpack_split_float 

Source
pub fn decode_modelpack_split_float<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 float 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.