pub fn decode_yolo_segdet_float<T>(
boxes: ArrayView2<'_, T>,
protos: ArrayView3<'_, T>,
score_threshold: f32,
iou_threshold: f32,
nms: Option<Nms>,
output_boxes: &mut Vec<DetectBox>,
output_masks: &mut Vec<Segmentation>,
) -> Result<(), DecoderError>Expand description
Decodes YOLO detection and segmentation outputs from float tensors into detection boxes and segmentation masks.
Boxes are expected to be in XYWH format.
Expected shapes of inputs:
- boxes: (4 + num_classes + num_protos, num_boxes)
- protos: (proto_height, proto_width, num_protos)
ยงErrors
Returns DecoderError::InvalidShape if bounding boxes are not normalized.