Referentie voor ultralytics/nn/modules/block.py
Opmerking
Dit bestand is beschikbaar op https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/nn/modules/block .py. Als je een probleem ziet, help het dan oplossen door een Pull Request 🛠️ bij te dragen. Bedankt 🙏!
ultralytics.nn.modules.block.DFL
Basis: Module
Integrale module van Distribution Focal Loss (DFL).
Voorgesteld in Gegeneraliseerd focaal verlies https://ieeexplore.ieee.org/document/9792391
Broncode in ultralytics/nn/modules/block.py
__init__(c1=16)
Initialiseer een convolutionele laag met een gegeven aantal invoerkanalen.
Broncode in ultralytics/nn/modules/block.py
forward(x)
Past een transformatorlaag toe op invoer tensor 'x' en retourneert een tensor.
ultralytics.nn.modules.block.Proto
Basis: Module
YOLOv8 Masker Proto-module voor segmentatiemodellen.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c_=256, c2=32)
Initialiseert de Proto-module YOLOv8 mask met een opgegeven aantal proto's en maskers.
Invoer argumenten zijn ch_in, aantal protos, aantal maskers.
Broncode in ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.HGStem
Basis: Module
Stamblok van PPHGNetV2 met 5 convoluties en één maxpool2d.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Broncode in ultralytics/nn/modules/block.py
__init__(c1, cm, c2)
Initialiseer de SPP-laag met invoer-/uitvoerkanalen en opgegeven kernelgroottes voor max pooling.
Broncode in ultralytics/nn/modules/block.py
forward(x)
Voorwaartse doorgang van een PPHGNetV2 backbone laag.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.HGBlock
Basis: Module
HG_Blok van PPHGNetV2 met 2 convoluties en LightConv.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Broncode in ultralytics/nn/modules/block.py
__init__(c1, cm, c2, k=3, n=6, lightconv=False, shortcut=False, act=nn.ReLU())
Initialiseert een CSP Bottleneck met 1 convolutie met behulp van opgegeven invoer- en uitvoerkanalen.
Broncode in ultralytics/nn/modules/block.py
forward(x)
Voorwaartse doorgang van een PPHGNetV2 backbone laag.
ultralytics.nn.modules.block.SPP
Basis: Module
Ruimtelijke piramide pooling (SPP) laag https://arxiv.org/abs/1406.4729.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, k=(5, 9, 13))
Initialiseer de SPP-laag met invoer-/uitvoerkanalen en pooling kernelgroottes.
Broncode in ultralytics/nn/modules/block.py
forward(x)
Voorwaartse passage van de SPP-laag, waarbij de ruimtelijke piramide wordt samengevoegd.
ultralytics.nn.modules.block.SPPF
Basis: Module
Spatial Pyramid Pooling - Fast (SPPF) laag voor YOLOv5 door Glenn Jocher.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, k=5)
Initialiseert de SPPF-laag met opgegeven in-/uitvoerkanalen en kernelgrootte.
Deze module is gelijkwaardig aan SPP(k=(5, 9, 13)).
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C1
Basis: Module
CSP Bottleneck met 1 convolutie.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1)
Initialiseert de CSP Bottleneck met configuraties voor 1 convolutie met argumenten ch_in, ch_out, getal.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2
Basis: Module
CSP Bottleneck met 2 convoluties.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Initialiseert de CSP Bottleneck met 2 convoluties module met argumenten ch_in, ch_out, aantal, snelkopp, groepen, uitbreiding.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2f
Basis: Module
Snellere implementatie van CSP Bottleneck met 2 convoluties.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=False, g=1, e=0.5)
Initialiseer de CSP-bottlenecklaag met twee convoluties met argumenten ch_in, ch_out, aantal, snelkoppeling, groepen, uitbreiding.
Broncode in ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
Forward pass met split() in plaats van chunk().
ultralytics.nn.modules.block.C3
Basis: Module
CSP Bottleneck met 3 convoluties.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Initialiseer de CSP Bottleneck met de gegeven kanalen, aantal, snelkoppeling, groepen en uitbreidingswaarden.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3x
Basis: C3
C3-module met kruisconvoluties.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
C3TR instantie initialiseren en standaardparameters instellen.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepC3
Basis: Module
Rep C3.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=3, e=1.0)
Initialiseer CSP Bottleneck met een enkele convolutie met invoerkanalen, uitvoerkanalen en aantal.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3TR
Basis: C3
C3 module met TransformerBlock().
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Initialiseer de module C3Ghost met GhostBottleneck().
ultralytics.nn.modules.block.C3Ghost
Basis: C3
C3 module met GhostBottleneck().
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Initialiseer de module 'SPP' met verschillende poolgroottes voor ruimtelijke piramidepooling.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.GhostBottleneck
Basis: Module
Ghost Bottleneck https://github.com/huawei-noah/ghostnet.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, k=3, s=1)
Initialiseert GhostBottleneck module met argumenten ch_in, ch_out, kernel, stride.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.Bottleneck
Basis: Module
Standaard knelpunt.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Initialiseert een knelpuntmodule met gegeven invoer-/uitvoerkanalen, snelkoppelingsoptie, groep, kernels en uitbreiding.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.BottleneckCSP
Basis: Module
CSP Knelpunt https://github.com/WongKinYiu/CrossStagePartialNetworks.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Initialiseert de CSP Bottleneck gegeven argumenten voor ch_in, ch_out, aantal, snelkoppeling, groepen, uitbreiding.
Broncode in ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.ResNetBlock
Basis: Module
ResNet blok met standaard convolutielagen.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, e=4)
Initialiseer convolutie met gegeven parameters.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ResNetLayer
Basis: Module
ResNet laag met meerdere ResNet blokken.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, is_first=False, n=1, e=4)
Initialiseert de ResNetLayer op basis van argumenten.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.MaxSigmoidAttnBlock
Basis: Module
Max Sigmoid aandachtsblok.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, nh=1, ec=128, gc=512, scale=False)
Initialiseert MaxSigmoidAttnBlock met opgegeven argumenten.
Broncode in ultralytics/nn/modules/block.py
forward(x, guide)
Voorwaarts proces.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2fAttn
Basis: Module
C2f module met een extra attn module.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, ec=128, nh=1, gc=512, shortcut=False, g=1, e=0.5)
Initialiseer de CSP-bottlenecklaag met twee convoluties met argumenten ch_in, ch_out, aantal, snelkoppeling, groepen, uitbreiding.
Broncode in ultralytics/nn/modules/block.py
forward(x, guide)
Voorwaartse doorgang door C2f-laag.
forward_split(x, guide)
Forward pass met split() in plaats van chunk().
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ImagePoolingAttn
Basis: Module
ImagePoolingAttn: Verbeter de tekstinbeddingen met beeldbewuste informatie.
Broncode in ultralytics/nn/modules/block.py
__init__(ec=256, ch=(), ct=512, nh=8, k=3, scale=False)
Initialiseert ImagePoolingAttn met opgegeven argumenten.
Broncode in ultralytics/nn/modules/block.py
forward(x, text)
Voert aandachtsmechanisme uit op invoer tensor x en gids tensor.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ContrastiveHead
Basis: Module
Contrastive Head for YOLO-World berekent de regio-tekstscores op basis van de overeenkomst tussen beeld- en tekstkenmerken. kenmerken.
Broncode in ultralytics/nn/modules/block.py
__init__()
Initialiseert ContrastiveHead met opgegeven regio-tekst overeenkomstparameters.
Broncode in ultralytics/nn/modules/block.py
forward(x, w)
Voorwaartse functie van contrastief leren.
ultralytics.nn.modules.block.BNContrastiveHead
Basis: Module
Batch Norm Contrastief Hoofd voor YOLO-Wereld met gebruik van batch norm in plaats van l2-normalisatie.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
embed_dims |
int
|
Afmetingen van tekst- en afbeeldingskenmerken insluiten. |
vereist |
Broncode in ultralytics/nn/modules/block.py
__init__(embed_dims)
Initialiseer ContrastiveHead met regio-tekst overeenkomstparameters.
Broncode in ultralytics/nn/modules/block.py
forward(x, w)
Voorwaartse functie van contrastief leren.
ultralytics.nn.modules.block.RepBottleneck
Basis: Bottleneck
Rep knelpunt.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Initialiseert een RepBottleneck module met aanpasbare in/uit kanalen, snelkoppelingsoptie, groepen en uitbreiding verhouding.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepCSP
Basis: C3
Rep CSP Bottleneck met 3 convoluties.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Initialiseert RepCSP laag met gegeven kanalen, herhalingen, snelkoppelingen, groepen en uitbreidingsratio.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepNCSPELAN4
Basis: Module
CSP-ELAN.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, c3, c4, n=1)
Initialiseert de CSP-ELAN laag met gespecificeerde kanaalgroottes, herhalingen en convoluties.
Broncode in ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
Forward pass met split() in plaats van chunk().
ultralytics.nn.modules.block.ADown
Basis: Module
Naar beneden.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2)
Initialiseert ADown module met convolutielagen om de invoer van kanalen c1 naar c2 te downsamplen.
Broncode in ultralytics/nn/modules/block.py
forward(x)
Voorwaartse doorgang door ADown laag.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.SPPELAN
Basis: Module
SPP-ELAN.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2, c3, k=5)
Initialiseert SPP-ELAN blok met convolutie en max pooling lagen voor ruimtelijke piramide pooling.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.Silence
Basis: Module
Stilte.
Broncode in ultralytics/nn/modules/block.py
__init__()
ultralytics.nn.modules.block.CBLinear
Basis: Module
CBLineair.
Broncode in ultralytics/nn/modules/block.py
__init__(c1, c2s, k=1, s=1, p=None, g=1)
Initialiseert de CBLinear module en geeft de invoer ongewijzigd door.
Broncode in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.CBFuse
Basis: Module
CBFuse.
Broncode in ultralytics/nn/modules/block.py
__init__(idx)
forward(xs)
Voorwaartse doorgang door CBFuse laag.