Struct RtmDetConfig 

pub struct RtmDetConfig {
    pub in_chans: usize,
    pub img_size: usize,
    pub stem_channels: usize,
    pub stage_channels: Vec<usize>,
    pub n_bottlenecks: usize,
    pub dw_kernel: usize,
    pub neck_channels: usize,
    pub n_classes: usize,
}

RTMDet hyper-parameters.

Fields

in_chans: usize

Input image channels (e.g. 3).

img_size: usize

Square input spatial size.

stem_channels: usize

Stem output channels.

stage_channels: Vec<usize>

Output channels of each backbone stage (each value must be even).

n_bottlenecks: usize

Bottlenecks per CSP layer.

dw_kernel: usize

Large depthwise kernel size (odd).

neck_channels: usize

Uniform PAFPN / head channel count.

n_classes: usize

Number of object classes.

Implementations

impl RtmDetConfig

pub fn new( in_chans: usize, img_size: usize, stem_channels: usize, stage_channels: Vec<usize>, n_bottlenecks: usize, dw_kernel: usize, neck_channels: usize, n_classes: usize, ) -> VisionResult<Self>

Create and validate a configuration.

Errors

• VisionError::InvalidImageSize if in_chans == 0 or img_size == 0.
• VisionError::EmptyInput if stage_channels is empty.
• VisionError::InvalidNumClasses if n_classes == 0.
• VisionError::InvalidPatchSize if dw_kernel is even or zero.
• VisionError::DimensionMismatch if any stage channel count is odd or zero, or neck_channels/stem_channels is zero.

pub fn tiny() -> Self

A tiny configuration for unit tests: 3×32×32 input, stem 8, stages [8, 16, 16], 1 bottleneck, 5×5 depthwise, neck 8, 4 classes.

pub fn n_levels(&self) -> usize

Number of pyramid levels (= number of backbone stages).

Trait Implementations

impl Clone for RtmDetConfig

fn clone(&self) -> RtmDetConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RtmDetConfig

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

Formats the value using the given formatter.

impl PartialEq for RtmDetConfig

fn eq(&self, other: &RtmDetConfig) -> 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 RtmDetConfig