Skip to main content

vision_rs/models/yolo/kernels/
mod.rs

1/// Attention kernels (flash attention, position-sensitive attention).
2pub mod attention;
3/// Kernel that decodes raw detection head output into boxes/scores/classes.
4pub mod detect_decode;
5/// Loss-related kernels (CIoU, classification loss).
6pub mod loss;