Conv
Bases: nn.Module
Standard convolution with args(ch_in, ch_out, kernel, stride, padding, groups, dilation, activation).
Source code in ultralytics/nn/modules/conv.py
__init__(c1, c2, k=1, s=1, p=None, g=1, d=1, act=True)
Initialize Conv layer with given arguments including activation.
Source code in ultralytics/nn/modules/conv.py
forward(x)
Conv2
Bases: Conv
Simplified RepConv module with Conv fusing.
Source code in ultralytics/nn/modules/conv.py
__init__(c1, c2, k=3, s=1, p=None, g=1, d=1, act=True)
Initialize Conv layer with given arguments including activation.
Source code in ultralytics/nn/modules/conv.py
forward(x)
fuse_convs()
Fuse parallel convolutions.
Source code in ultralytics/nn/modules/conv.py
LightConv
Bases: nn.Module
Light convolution with args(ch_in, ch_out, kernel). https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Source code in ultralytics/nn/modules/conv.py
__init__(c1, c2, k=1, act=nn.ReLU())
Initialize Conv layer with given arguments including activation.
DWConv
DWConvTranspose2d
Bases: nn.ConvTranspose2d
Depth-wise transpose convolution.
Source code in ultralytics/nn/modules/conv.py
ConvTranspose
Bases: nn.Module
Convolution transpose 2d layer.
Source code in ultralytics/nn/modules/conv.py
__init__(c1, c2, k=2, s=2, p=0, bn=True, act=True)
Initialize ConvTranspose2d layer with batch normalization and activation function.
Source code in ultralytics/nn/modules/conv.py
forward(x)
Focus
Bases: nn.Module
Focus wh information into c-space.
Source code in ultralytics/nn/modules/conv.py
GhostConv
Bases: nn.Module
Ghost Convolution https://github.com/huawei-noah/ghostnet.
Source code in ultralytics/nn/modules/conv.py
forward(x)
RepConv
Bases: nn.Module
RepConv is a basic rep-style block, including training and deploy status This code is based on https://github.com/DingXiaoH/RepVGG/blob/main/repvgg.py
Source code in ultralytics/nn/modules/conv.py
144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 |
|
forward(x)
ChannelAttention
Bases: nn.Module
Channel-attention module https://github.com/open-mmlab/mmdetection/tree/v3.0.0rc1/configs/rtmdet.
Source code in ultralytics/nn/modules/conv.py
SpatialAttention
Bases: nn.Module
Spatial-attention module.
Source code in ultralytics/nn/modules/conv.py
__init__(kernel_size=7)
Initialize Spatial-attention module with kernel size argument.
Source code in ultralytics/nn/modules/conv.py
forward(x)
Apply channel and spatial attention on input for feature recalibration.
CBAM
Bases: nn.Module
Convolutional Block Attention Module.
Source code in ultralytics/nn/modules/conv.py
Concat
Bases: nn.Module
Concatenate a list of tensors along dimension.
Source code in ultralytics/nn/modules/conv.py
__init__(dimension=1)
autopad
Created 2023-05-11, Updated 2023-05-28
Authors: Glenn Jocher (4)