Skip to main content

c3k2_psa

vision_rs::models::yolo::yolo26::blocks::c3k2

Function c3k2_psa 

Source 
pub fn c3k2_psa<D: Float + Send + Sync + 'static>(
    c_in: usize,
    c_out: usize,
    n: usize,
    shortcut: bool,
    e: f32,
) -> impl Fn(SymTensor) -> SymTensor
Expand description

C3k2 variant whose inner blocks are Sequential([Bottleneck, PSABlock]).

Used for model.22 (P5/32 detect-path block) in YOLO26. The inner channel width c = c_out * e; num_heads = c / 64, key_dim = 32 follow the same convention as C2PSA.