Struct SamConfig 

pub struct SamConfig {
    pub in_chans: usize,
    pub img_size: usize,
    pub patch_size: usize,
    pub embed_dim: usize,
    pub enc_depth: usize,
    pub enc_heads: usize,
    pub enc_mlp_ratio: usize,
    pub dec_depth: usize,
    pub dec_heads: usize,
    pub dec_mlp_dim: usize,
    pub n_mask: usize,
}

SAM hyper-parameters.

Fields

in_chans: usize

Input image channels.

img_size: usize

Square input size.

patch_size: usize

Patch size of the image encoder.

embed_dim: usize

Embedding dimension (shared across encoder / prompt / decoder).

enc_depth: usize

Encoder transformer depth.

enc_heads: usize

Encoder attention heads.

enc_mlp_ratio: usize

Encoder MLP ratio.

dec_depth: usize

Decoder two-way transformer depth.

dec_heads: usize

Decoder attention heads.

dec_mlp_dim: usize

Decoder MLP hidden dimension.

n_mask: usize

Number of output mask tokens.

Implementations

impl SamConfig

pub fn new( in_chans: usize, img_size: usize, patch_size: usize, embed_dim: usize, enc_depth: usize, enc_heads: usize, enc_mlp_ratio: usize, dec_depth: usize, dec_heads: usize, dec_mlp_dim: usize, n_mask: usize, ) -> VisionResult<Self>

Create and validate a configuration.

Errors

• VisionError::InvalidEmbedDim if embed_dim is 0 or odd.
• VisionError::InvalidPatchSize if patch_size does not divide img_size.
• VisionError::HeadDimMismatch if a head count does not divide embed_dim.
• VisionError::EmptyInput if n_mask == 0.

pub fn tiny() -> Self

A tiny configuration for unit tests: 3×32×32 input, patch 8, embed 16, 2 encoder blocks (2 heads), 2-deep decoder (2 heads), 3 masks.

Trait Implementations

impl Clone for SamConfig

fn clone(&self) -> SamConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SamConfig

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

Formats the value using the given formatter.

impl PartialEq for SamConfig

fn eq(&self, other: &SamConfig) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for SamConfig