| |
| """ |
| Ultralytics modules. |
| |
| Example: |
| Visualize a module with Netron. |
| ```python |
| from ultralytics.nn.modules import * |
| import torch |
| import os |
| |
| x = torch.ones(1, 128, 40, 40) |
| m = Conv(128, 128) |
| f = f'{m._get_name()}.onnx' |
| torch.onnx.export(m, x, f) |
| os.system(f'onnxslim {f} {f} && open {f}') # pip install onnxslim |
| ``` |
| """ |
|
|
| from .block import ( |
| C1, |
| C2, |
| C3, |
| C3TR, |
| DFL, |
| SPP, |
| SPPF, |
| Bottleneck, |
| BottleneckCSP, |
| C2f, |
| C2fAttn, |
| ImagePoolingAttn, |
| C3Ghost, |
| C3x, |
| GhostBottleneck, |
| HGBlock, |
| HGStem, |
| Proto, |
| RepC3, |
| ResNetLayer, |
| ContrastiveHead, |
| BNContrastiveHead, |
| RepNCSPELAN4, |
| ADown, |
| SPPELAN, |
| CBFuse, |
| CBLinear, |
| Silence, |
| PSA, |
| C2fCIB, |
| SCDown, |
| RepVGGDW |
| ) |
| from .conv import ( |
| CBAM, |
| ChannelAttention, |
| Concat, |
| Conv, |
| Conv2, |
| ConvTranspose, |
| DWConv, |
| DWConvTranspose2d, |
| Focus, |
| GhostConv, |
| LightConv, |
| RepConv, |
| SpatialAttention, |
| ) |
| from .head import OBB, Classify, Detect, Pose, RTDETRDecoder, Segment, WorldDetect, v10Detect |
| from .transformer import ( |
| AIFI, |
| MLP, |
| DeformableTransformerDecoder, |
| DeformableTransformerDecoderLayer, |
| LayerNorm2d, |
| MLPBlock, |
| MSDeformAttn, |
| TransformerBlock, |
| TransformerEncoderLayer, |
| TransformerLayer, |
| ) |
|
|
| __all__ = ( |
| "Conv", |
| "Conv2", |
| "LightConv", |
| "RepConv", |
| "DWConv", |
| "DWConvTranspose2d", |
| "ConvTranspose", |
| "Focus", |
| "GhostConv", |
| "ChannelAttention", |
| "SpatialAttention", |
| "CBAM", |
| "Concat", |
| "TransformerLayer", |
| "TransformerBlock", |
| "MLPBlock", |
| "LayerNorm2d", |
| "DFL", |
| "HGBlock", |
| "HGStem", |
| "SPP", |
| "SPPF", |
| "C1", |
| "C2", |
| "C3", |
| "C2f", |
| "C2fAttn", |
| "C3x", |
| "C3TR", |
| "C3Ghost", |
| "GhostBottleneck", |
| "Bottleneck", |
| "BottleneckCSP", |
| "Proto", |
| "Detect", |
| "Segment", |
| "Pose", |
| "Classify", |
| "TransformerEncoderLayer", |
| "RepC3", |
| "RTDETRDecoder", |
| "AIFI", |
| "DeformableTransformerDecoder", |
| "DeformableTransformerDecoderLayer", |
| "MSDeformAttn", |
| "MLP", |
| "ResNetLayer", |
| "OBB", |
| "WorldDetect", |
| "ImagePoolingAttn", |
| "ContrastiveHead", |
| "BNContrastiveHead", |
| "RepNCSPELAN4", |
| "ADown", |
| "SPPELAN", |
| "CBFuse", |
| "CBLinear", |
| "Silence", |
| "PSA", |
| "C2fCIB", |
| "SCDown", |
| "RepVGGDW", |
| "v10Detect" |
| ) |
|
|
