Skip to main content

decode_yolo_split_end_to_end_segdet_float

edgefirst_decoder::yolo

Function decode_yolo_split_end_to_end_segdet_float 

Source 
pub fn decode_yolo_split_end_to_end_segdet_float<T>(
    boxes: ArrayView2<'_, T>,
    scores: ArrayView2<'_, T>,
    classes: ArrayView2<'_, T>,
    mask_coeff: ArrayView2<'_, T>,
    protos: ArrayView3<'_, T>,
    score_threshold: f32,
    output_boxes: &mut Vec<DetectBox>,
    output_masks: &mut Vec<Segmentation>,
) -> Result<(), DecoderError>
where
    T: Float + AsPrimitive<f32> + Send + Sync + 'static,
    f32: AsPrimitive<T>,
Expand description

Decodes split end-to-end YOLO detection + segmentation outputs.

Input shapes (after batch dim removed):

  • boxes: (4, N) — xyxy pixel coordinates
  • scores: (1, N) — confidence
  • classes: (1, N) — class index
  • mask_coeff: (num_protos, N) — mask coefficients per detection
  • protos: (proto_h, proto_w, num_protos) — prototype masks