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,
},
}Variants§
ModelPackSegDet
ModelPackSegDetSplit
ModelPackDet
ModelPackDetSplit
ModelPackSeg
Fields
§
segmentation: SegmentationYoloDet
YoloSegDet
YoloSplitDet
YoloSplitSegDet
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]
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
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
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>
Converts
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>
Converts
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