Struct DetrConfig 

pub struct DetrConfig {
    pub n_queries: usize,
    pub embed_dim: usize,
    pub n_heads: usize,
    pub depth: usize,
    pub mlp_ratio: usize,
}

DETR decoder hyper-parameters.

Fields

n_queries: usize

Number of object query vectors.

embed_dim: usize

Embedding dimension for all tokens (queries and encoder features).

n_heads: usize

Number of attention heads (must divide embed_dim).

depth: usize

Number of decoder layers.

mlp_ratio: usize

MLP expansion factor: mlp_dim = mlp_ratio * embed_dim.

Implementations

impl DetrConfig

pub fn new( n_queries: usize, embed_dim: usize, n_heads: usize, depth: usize, mlp_ratio: usize, ) -> VisionResult<Self>

Construct a validated DetrConfig.

Errors

• InvalidEmbedDim if embed_dim == 0.
• InvalidNumHeads if n_heads == 0.
• HeadDimMismatch if embed_dim % n_heads != 0.
• DimensionMismatch if n_queries == 0, depth == 0, or mlp_ratio == 0.

pub fn tiny() -> Self

A tiny configuration for unit tests.

n_queries=4, embed_dim=32, n_heads=4, depth=1, mlp_ratio=4.

pub fn mlp_dim(&self) -> usize

MLP hidden dimension.

pub fn head_dim(&self) -> usize

Per-head dimension.

Trait Implementations

impl Clone for DetrConfig

fn clone(&self) -> DetrConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DetrConfig

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

Formats the value using the given formatter.