vision_rs/models/yolo/yolo26/blocks/mod.rs
1/*
2 * Copyright 2026 Teenygrad
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17
18/// Residual bottleneck block.
19pub mod bottleneck;
20/// C2PSA block: cross-stage partial block with position-sensitive attention.
21pub mod c2psa;
22/// C3k2 block: CSP bottleneck variant used throughout the backbone/neck.
23pub mod c3k2;
24/// Channel-wise concatenation of feature maps.
25pub mod concat;
26/// Conv + BatchNorm + activation block.
27pub mod conv;
28/// Detection head block(s).
29pub mod detect;
30/// Spatial Pyramid Pooling - Fast block.
31pub mod sppf;
32/// Nearest-neighbor upsampling block.
33pub mod upsample;