Struct darknet::Network

pub struct Network { /* fields omitted */ }

Methods

impl Network

pub fn load(
    cfg: &str,
    weights: Option<&str>,
    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(&mut self, file_name: &str)

Save network weights to file

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

Returns vector of object labels