Enum ModelType 

pub enum ModelType {
    ModelPackSegDet {
        boxes: Boxes,
        scores: Scores,
        segmentation: Segmentation,
    },
    ModelPackSegDetSplit {
        detection: Vec<Detection>,
        segmentation: Segmentation,
    },
    ModelPackDet {
        boxes: Boxes,
        scores: Scores,
    },
    ModelPackDetSplit {
        detection: Vec<Detection>,
    },
    ModelPackSeg {
        segmentation: Segmentation,
    },
    YoloDet {
        boxes: Detection,
    },
    YoloSegDet {
        boxes: Detection,
        protos: Protos,
    },
    YoloSplitDet {
        boxes: Boxes,
        scores: Scores,
    },
    YoloSplitSegDet {
        boxes: Boxes,
        scores: Scores,
        mask_coeff: MaskCoefficients,
        protos: Protos,
    },
    YoloEndToEndDet {
        boxes: Detection,
    },
    YoloEndToEndSegDet {
        boxes: Detection,
        protos: Protos,
    },
    YoloSplitEndToEndDet {
        boxes: Boxes,
        scores: Scores,
        classes: Classes,
    },
    YoloSplitEndToEndSegDet {
        boxes: Boxes,
        scores: Scores,
        classes: Classes,
        mask_coeff: MaskCoefficients,
        protos: Protos,
    },
}

Variants

ModelPackSegDet

Fields

boxes: Boxes

scores: Scores

segmentation: Segmentation

ModelPackSegDetSplit

Fields

detection: Vec<Detection>

segmentation: Segmentation

ModelPackDet

Fields

boxes: Boxes

scores: Scores

ModelPackDetSplit

Fields

detection: Vec<Detection>

ModelPackSeg

Fields

segmentation: Segmentation

YoloDet

Fields

boxes: Detection

YoloSegDet

Fields

boxes: Detection

protos: Protos

YoloSplitDet

Fields

boxes: Boxes

scores: Scores

YoloSplitSegDet

Fields

boxes: Boxes

scores: Scores

mask_coeff: MaskCoefficients

protos: Protos

YoloEndToEndDet

End-to-end YOLO detection (post-NMS output from model) Input shape: (1, N, 6+) where columns are [x1, y1, x2, y2, conf, class, …]

Fields

boxes: Detection

YoloEndToEndSegDet

End-to-end YOLO detection + segmentation (post-NMS output from model) Input shape: (1, N, 6 + num_protos) where columns are [x1, y1, x2, y2, conf, class, mask_coeff_0, …, mask_coeff_31]

Fields

boxes: Detection

protos: Protos

YoloSplitEndToEndDet

Split end-to-end YOLO detection (onnx2tf splits [1,N,6] into 3 tensors) boxes: [batch, N, 4] xyxy, scores: [batch, N, 1], classes: [batch, N, 1]

Fields

boxes: Boxes

scores: Scores

classes: Classes

YoloSplitEndToEndSegDet

Split end-to-end YOLO seg detection (onnx2tf splits into 5 tensors)

Fields

boxes: Boxes

scores: Scores

classes: Classes

mask_coeff: MaskCoefficients

protos: Protos

Trait Implementations

impl Clone for ModelType

fn clone(&self) -> ModelType

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ModelType

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for ModelType

fn eq(&self, other: &ModelType) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for ModelType