Struct darknet::Network

pub struct Network { /* fields omitted */ }

Methods

impl Network

pub fn load<C: AsRef<Path> + ?Sized, W: AsRef<Path> + ?Sized>(
    cfg: &C,
    weights: Option<&W>,
    clear: bool,
    labels: Vec<String>
) -> Option<Network>

Load network from config file cfg (under cfg/ subdir) and weights file weights (can be obtained from https://pjreddie.com/darknet/, optional if training).
clear - Reset network data (used for training).
labels - vector of object labels the model was trained on (i.e. vec!["car", "bird", "dog"...]).

pub fn get_w(&self) -> usize

Network input width.

pub fn get_h(&self) -> usize

Network input height.

pub fn predict(
    &mut self,
    image: &mut Image,
    thresh: f32,
    nms: f32
) -> Detections

Predict and return object bboxes (with probability > 'thresh').
'nms' - overlap threshold for non-maximum suppression (higher = more overlapping allowed)

pub fn save_weights<P: AsRef<Path> + ?Sized>(&mut self, file_name: &P)

Save network weights to file

pub fn get_labels(&self) -> Vec<String>

Returns vector of object labels

Trait Implementations

impl Drop for Network

fn drop(&mut self)

Executes the destructor for this type.