Enum VisionTaskConfig 

pub enum VisionTaskConfig {
    Classification {
        num_classes: usize,
        multi_label: bool,
    },
    ObjectDetection {
        num_classes: usize,
        anchor_sizes: Vec<(usize, usize)>,
        iou_threshold: f64,
    },
    Segmentation {
        num_classes: usize,
        output_stride: usize,
    },
    InstanceSegmentation {
        num_classes: usize,
        mask_resolution: (usize, usize),
    },
    FeatureExtraction {
        feature_dim: usize,
        normalize: bool,
    },
    Generation {
        latent_dim: usize,
        output_channels: usize,
    },
}

Vision task configurations

Variants

Classification

Image classification

Fields

num_classes: usize

multi_label: bool

ObjectDetection

Object detection

Fields

num_classes: usize

anchor_sizes: Vec<(usize, usize)>

iou_threshold: f64

Segmentation

Semantic segmentation

Fields

num_classes: usize

output_stride: usize

InstanceSegmentation

Instance segmentation

Fields

num_classes: usize

mask_resolution: (usize, usize)

FeatureExtraction

Feature extraction

Fields

feature_dim: usize

normalize: bool

Generation

Image generation

Fields

latent_dim: usize

output_channels: usize

Trait Implementations

impl Clone for VisionTaskConfig

fn clone(&self) -> VisionTaskConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for VisionTaskConfig

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

Formats the value using the given formatter.