Riferimento per ultralytics/nn/modules/block.py
Nota
Questo file è disponibile su https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/nn/modules/block .py. Se noti un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.nn.modules.block.DFL
Basi: Module
Modulo integrale di distribuzione della perdita focale (DFL).
Proposto in Perdita Focale Generalizzata https://ieeexplore.ieee.org/document/9792391
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1=16)
Inizializza uno strato convoluzionale con un determinato numero di canali di ingresso.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Applica un livello di trasformazione all'ingresso tensor 'x' e restituisce un tensor.
ultralytics.nn.modules.block.Proto
Basi: Module
YOLOv8 maschera Modulo Proto per i modelli di segmentazione.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c_=256, c2=32)
Inizializza il modulo Proto della maschera YOLOv8 con il numero specificato di proto e maschere.
Gli argomenti in ingresso sono ch_in, numero di proto, numero di maschere.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Esegue un passaggio in avanti attraverso i livelli utilizzando un'immagine di ingresso sovracampionata.
ultralytics.nn.modules.block.HGStem
Basi: Module
StemBlock di PPHGNetV2 con 5 convoluzioni e un maxpool2d.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, cm, c2)
Inizializza il livello SPP con i canali di ingresso/uscita e le dimensioni del kernel specificate per il pooling massimo.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Passaggio in avanti di un livello di backbone PPHGNetV2.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.HGBlock
Basi: Module
HG_Block di PPHGNetV2 con 2 convoluzioni e LightConv.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, cm, c2, k=3, n=6, lightconv=False, shortcut=False, act=nn.ReLU())
Inizializza un CSP Bottleneck con 1 convoluzione utilizzando i canali di ingresso e di uscita specificati.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Passaggio in avanti di un livello di backbone PPHGNetV2.
ultralytics.nn.modules.block.SPP
Basi: Module
Strato Spatial Pyramid Pooling (SPP) https://arxiv.org/abs/1406.4729.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, k=(5, 9, 13))
Inizializza il livello SPP con i canali di ingresso/uscita e le dimensioni del kernel di pooling.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Passaggio in avanti del livello SPP, che esegue il pooling della piramide spaziale.
ultralytics.nn.modules.block.SPPF
Basi: Module
Livello Spatial Pyramid Pooling - Fast (SPPF) per YOLOv5 di Glenn Jocher.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, k=5)
Inizializza il livello SPPF con i canali di ingresso/uscita e la dimensione del kernel indicati.
Questo modulo è equivalente a SPP(k=(5, 9, 13)).
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.C1
Basi: Module
Collo di bottiglia CSP con 1 convoluzione.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1)
Inizializza il CSP Bottleneck con le configurazioni per 1 convoluzione con gli argomenti ch_in, ch_out, numero.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2
Basi: Module
Collo di bottiglia CSP con 2 convoluzioni.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inizializza il modulo CSP Bottleneck with 2 convolutions con gli argomenti ch_in, ch_out, number, shortcut, gruppi, espansione.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Passaggio in avanti attraverso il collo di bottiglia del CSP con 2 convoluzioni.
ultralytics.nn.modules.block.C2f
Basi: Module
Implementazione più veloce del collo di bottiglia CSP con 2 convoluzioni.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=False, g=1, e=0.5)
Inizializza lo strato CSP a collo di bottiglia con due convoluzioni con argomenti ch_in, ch_out, numero, scorciatoia, gruppi, espansione.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
Passaggio in avanti usando split() invece di chunk().
ultralytics.nn.modules.block.C3
Basi: Module
Collo di bottiglia CSP con 3 convoluzioni.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inizializza il CSP Bottleneck con i canali, il numero, la scorciatoia, i gruppi e i valori di espansione indicati.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.C3x
Basi: C3
Modulo C3 con evoluzioni incrociate.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inizializza l'istanza C3TR e imposta i parametri predefiniti.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepC3
Basi: Module
Rep C3.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=3, e=1.0)
Inizializza il CSP Bottleneck con una singola convoluzione utilizzando i canali di ingresso, i canali di uscita e il numero.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3TR
Basi: C3
Modulo C3 con TransformerBlock().
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inizializza il modulo C3Ghost con GhostBottleneck().
ultralytics.nn.modules.block.C3Ghost
Basi: C3
Modulo C3 con GhostBottleneck().
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inizializza il modulo 'SPP' con varie dimensioni di pooling per il pooling piramidale spaziale.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.GhostBottleneck
Basi: Module
Collo di bottiglia fantasma https://github.com/huawei-noah/ghostnet.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, k=3, s=1)
Inizializza il modulo GhostBottleneck con gli argomenti ch_in, ch_out, kernel, stride.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.Bottleneck
Basi: Module
Collo di bottiglia standard.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Inizializza un modulo a collo di bottiglia con i canali di ingresso/uscita, l'opzione di scelta rapida, il gruppo, i kernel e l'espansione. espansione.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.BottleneckCSP
Basi: Module
Collo di bottiglia del CSP https://github.com/WongKinYiu/CrossStagePartialNetworks.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inizializza il CSP Bottleneck con gli argomenti ch_in, ch_out, numero, scorciatoia, gruppi, espansione.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Applica un collo di bottiglia CSP con 3 convoluzioni.
ultralytics.nn.modules.block.ResNetBlock
Basi: Module
Blocco ResNet con strati di convoluzione standard.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, e=4)
Inizializza la convoluzione con i parametri indicati.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ResNetLayer
Basi: Module
Strato ResNet con più blocchi ResNet.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, is_first=False, n=1, e=4)
Inizializza il ResNetLayer con i suoi argomenti.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.MaxSigmoidAttnBlock
Basi: Module
Blocco dell'attenzione max sigmoidea.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, nh=1, ec=128, gc=512, scale=False)
Inizializza MaxSigmoidAttnBlock con gli argomenti specificati.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x, guide)
Processo di avanzamento.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2fAttn
Basi: Module
Modulo C2f con un modulo attn aggiuntivo.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, ec=128, nh=1, gc=512, shortcut=False, g=1, e=0.5)
Inizializza lo strato CSP a collo di bottiglia con due convoluzioni con argomenti ch_in, ch_out, numero, scorciatoia, gruppi, espansione.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x, guide)
Passaggio in avanti attraverso lo strato C2f.
forward_split(x, guide)
Passaggio in avanti usando split() invece di chunk().
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ImagePoolingAttn
Basi: Module
ImagePoolingAttn: migliora le incorporazioni di testo con informazioni consapevoli delle immagini.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(ec=256, ch=(), ct=512, nh=8, k=3, scale=False)
Inizializza ImagePoolingAttn con gli argomenti specificati.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x, text)
Esegue il meccanismo di attenzione su tensor x e tensor guida.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ContrastiveHead
Basi: Module
Contrastive Head for YOLO-World calcola i punteggi regione-testo in base alla somiglianza tra le caratteristiche dell'immagine e del testo. caratteristiche.
Codice sorgente in ultralytics/nn/modules/block.py
__init__()
Inizializza ContrastiveHead con i parametri di somiglianza regione-testo specificati.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x, w)
La funzione di apprendimento contrastivo.
ultralytics.nn.modules.block.BNContrastiveHead
Basi: Module
Batch Norm Contrastive Head per YOLO-World utilizzando la batch norm invece della normalizzazione l2.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
embed_dims |
int
|
Incorpora le dimensioni delle caratteristiche di testo e immagine. |
richiesto |
Codice sorgente in ultralytics/nn/modules/block.py
__init__(embed_dims)
Inizializza ContrastiveHead con i parametri di somiglianza regione-testo.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x, w)
La funzione di apprendimento contrastivo.
ultralytics.nn.modules.block.RepBottleneck
Basi: Bottleneck
Collo di bottiglia delle rep.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Inizializza un modulo RepBottleneck con canali in/out personalizzabili, opzione di scelta rapida, gruppi e rapporto di espansione. rapporto.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepCSP
Basi: C3
Rep CSP Bottleneck con 3 convoluzioni.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inizializza il livello RepCSP con i canali, le ripetizioni, le scorciatoie, i gruppi e il rapporto di espansione indicati.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepNCSPELAN4
Basi: Module
CSP-ELAN.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, c3, c4, n=1)
Inizializza il livello CSP-ELAN con le dimensioni dei canali, le ripetizioni e le convoluzioni specificate.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Passaggio in avanti attraverso il livello RepNCSPELAN4.
forward_split(x)
Passaggio in avanti usando split() invece di chunk().
ultralytics.nn.modules.block.ADown
Basi: Module
ADown.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2)
Inizializza il modulo ADown con gli strati di convoluzione per ricampionare l'ingresso dai canali c1 e c2.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
Passaggio in avanti attraverso il livello ADown.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.SPPELAN
Basi: Module
SPP-ELAN.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2, c3, k=5)
Inizializza il blocco SPP-ELAN con strati di convoluzione e pooling massimo per il pooling della piramide spaziale.
Codice sorgente in ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.Silence
Basi: Module
Silenzio.
Codice sorgente in ultralytics/nn/modules/block.py
__init__()
ultralytics.nn.modules.block.CBLinear
Basi: Module
CBLinear.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(c1, c2s, k=1, s=1, p=None, g=1)
Inizializza il modulo CBLinear, passando gli input invariati.
Codice sorgente in ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.CBFuse
Basi: Module
CBFuse.
Codice sorgente in ultralytics/nn/modules/block.py
__init__(idx)
Inizializza il modulo CBFuse con l'indice del livello per la fusione selettiva delle caratteristiche.
forward(xs)
Passaggio in avanti attraverso lo strato CBFuse.