Skip to main content

impl_yolo_split_segdet_float_proto

edgefirst_decoder::yolo

Function impl_yolo_split_segdet_float_proto 

Source 
pub fn impl_yolo_split_segdet_float_proto<B: BBoxTypeTrait, BOX: Float + AsPrimitive<f32> + Send + Sync, SCORE: Float + AsPrimitive<f32> + Send + Sync, MASK: Float + AsPrimitive<f32> + Send + Sync, PROTO: Float + AsPrimitive<f32> + Send + Sync>(
    boxes_tensor: ArrayView2<'_, BOX>,
    scores_tensor: ArrayView2<'_, SCORE>,
    mask_tensor: ArrayView2<'_, MASK>,
    protos: ArrayView3<'_, PROTO>,
    score_threshold: f32,
    iou_threshold: f32,
    nms: Option<Nms>,
    output_boxes: &mut Vec<DetectBox>,
) -> ProtoData
where
    f32: AsPrimitive<SCORE>,
Expand description

Proto-extraction variant of impl_yolo_split_segdet_float. Runs NMS but returns raw ProtoData instead of materialized masks.