pub enum ModelType {
Show 15 variants
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,
},
YoloSegDet2Way {
boxes: Detection,
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,
},
PerScale,
}Variants§
ModelPackSegDet
ModelPackSegDetSplit
ModelPackDet
ModelPackDetSplit
ModelPackSeg
Fields
segmentation: SegmentationYoloDet
YoloSegDet
YoloSplitDet
YoloSplitSegDet
YoloSegDet2Way
2-way split YOLO segmentation detection. Combined detection tensor (boxes + scores) with separate mask coefficients and prototype masks.
- detection: [1, nc+4, N] — boxes and scores combined
- mask_coeff: [1, 32, N] — mask coefficients (separate tensor)
- protos: [1, H/4, W/4, 32] — prototype masks
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, …]
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]
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]
YoloSplitEndToEndSegDet
Split end-to-end YOLO seg detection (onnx2tf splits into 5 tensors)
PerScale
Per-scale (physical-output-decomposition) YOLO model. The
per-scale subsystem (crates/decoder/src/per_scale/) owns
model decoding entirely; this variant exists as a marker so the
Decoder::model_type field has a sensible value for per-scale
Decoders that bypass the legacy ModelType-driven dispatch.
Trait Implementations§
impl StructuralPartialEq for ModelType
Auto Trait Implementations§
impl Freeze for ModelType
impl RefUnwindSafe for ModelType
impl Send for ModelType
impl Sync for ModelType
impl Unpin for ModelType
impl UnsafeUnpin for ModelType
impl UnwindSafe for ModelType
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more