2.1 KiB
| description |
|---|
| Explore Ultralytics neural network modules for convolution, attention, detection, pose, and classification in PyTorch. |
Conv
:::ultralytics.nn.modules.Conv
DWConv
:::ultralytics.nn.modules.DWConv
DWConvTranspose2d
:::ultralytics.nn.modules.DWConvTranspose2d
ConvTranspose
:::ultralytics.nn.modules.ConvTranspose
DFL
:::ultralytics.nn.modules.DFL
TransformerLayer
:::ultralytics.nn.modules.TransformerLayer
TransformerBlock
:::ultralytics.nn.modules.TransformerBlock
Bottleneck
:::ultralytics.nn.modules.Bottleneck
BottleneckCSP
:::ultralytics.nn.modules.BottleneckCSP
C3
:::ultralytics.nn.modules.C3
C2
:::ultralytics.nn.modules.C2
C2f
:::ultralytics.nn.modules.C2f
ChannelAttention
:::ultralytics.nn.modules.ChannelAttention
SpatialAttention
:::ultralytics.nn.modules.SpatialAttention
CBAM
:::ultralytics.nn.modules.CBAM
C1
:::ultralytics.nn.modules.C1
C3x
:::ultralytics.nn.modules.C3x
C3TR
:::ultralytics.nn.modules.C3TR
C3Ghost
:::ultralytics.nn.modules.C3Ghost
SPP
:::ultralytics.nn.modules.SPP
SPPF
:::ultralytics.nn.modules.SPPF
Focus
:::ultralytics.nn.modules.Focus
GhostConv
:::ultralytics.nn.modules.GhostConv
GhostBottleneck
:::ultralytics.nn.modules.GhostBottleneck
Concat
:::ultralytics.nn.modules.Concat
Proto
:::ultralytics.nn.modules.Proto
Ensemble
:::ultralytics.nn.modules.Ensemble
Detect
:::ultralytics.nn.modules.Detect
MLPBlock
:::ultralytics.nn.modules.MLPBlock
LayerNorm2d
:::ultralytics.nn.modules.LayerNorm2d
Segment
:::ultralytics.nn.modules.Segment
Pose
:::ultralytics.nn.modules.Pose
Classify
:::ultralytics.nn.modules.Classify
autopad
:::ultralytics.nn.modules.autopad