Skip to main content

postprocess_boxes_index_float

edgefirst_decoder::float

Function postprocess_boxes_index_float 

Source 
pub fn postprocess_boxes_index_float<B: BBoxTypeTrait, BOX: Float + AsPrimitive<f32> + Send + Sync, SCORE: Float + AsPrimitive<f32> + Send + Sync>(
    threshold: SCORE,
    boxes: ArrayView2<'_, BOX>,
    scores: ArrayView2<'_, SCORE>,
) -> Vec<(DetectBox, usize)>
Expand description

Post processes boxes and scores tensors into detection boxes, filtering out any boxes below the score threshold. The boxes tensor is converted to XYXY using the given BBoxTypeTrait. The order of the boxes is preserved.

This function is very similar to postprocess_boxes_float but will also return the index of the box. The boxes will be in ascending index order.