Struct YoloModelSession 

pub struct YoloModelSession {
    pub session: Session,
    pub labels: Vec<ArcStr>,
    pub probability_threshold: Option<f32>,
    pub iou_threshold: Option<f32>,
}

The YOLO model.

It is a wrapper around the ONNX runtime session and the YOLO labels.

Fields

session: Session

labels: Vec<ArcStr>

probability_threshold: Option<f32>

iou_threshold: Option<f32>

Implementations

impl YoloModelSession

pub fn new( session: Session, labels: impl Iterator<Item = impl Into<ArcStr>>, ) -> Self

Wrap a ONNX session to a YoloModelSession.

The session is the ONNX runtime session, and the labels are the YOLO labels.

pub fn new_v8(session: Session) -> Self

Wrap a ONNX session to a YoloModelSession based on the labels of YOLO v8 (v11).

pub fn from_filename_v8(filename: impl AsRef<Path>) -> Result<Self, YoloError>

Load the YOLO ONNX model from a filename.

You can use this function to load a YOLO v8 (v11) model from a file. The filename is the path to the ONNX model file.

You can export the ONNX model file according to Ultralytics’ manual.

pub fn get_labels(&self) -> &[ArcStr]

pub fn get_probability_threshold(&self) -> f32

pub fn get_iou_threshold(&self) -> f32

Trait Implementations

impl AsMut<Session> for YoloModelSession

fn as_mut(&mut self) -> &mut Session

Converts this type into a mutable reference of the (usually inferred) input type.

impl AsRef<Session> for YoloModelSession

fn as_ref(&self) -> &Session

Converts this type into a shared reference of the (usually inferred) input type.

impl Debug for YoloModelSession

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

Formats the value using the given formatter.