Skip to main content

impl_yolo_split_segdet_quant_proto

edgefirst_decoder::yolo

Function impl_yolo_split_segdet_quant_proto 

Source 
pub fn impl_yolo_split_segdet_quant_proto<B: BBoxTypeTrait, BOX: PrimInt + AsPrimitive<f32> + Send + Sync, SCORE: PrimInt + AsPrimitive<f32> + Send + Sync, MASK: PrimInt + AsPrimitive<f32> + Send + Sync, PROTO: PrimInt + AsPrimitive<f32> + AsPrimitive<i8> + Send + Sync>(
    boxes: (ArrayView2<'_, BOX>, Quantization),
    scores: (ArrayView2<'_, SCORE>, Quantization),
    mask_coeff: (ArrayView2<'_, MASK>, Quantization),
    protos: (ArrayView3<'_, PROTO>, Quantization),
    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_quant. Runs NMS but returns raw ProtoData instead of materialized masks.