Referencia para ultralytics/nn/modules/block.py
Nota
Este archivo está disponible en https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/nn/modules/block .py. Si detectas algún problema, por favor, ayuda a solucionarlo contribuyendo con una Pull Request 🛠️. ¡Gracias 🙏!
ultralytics.nn.modules.block.DFL
Bases: Module
Módulo integral de Pérdida Focal de Distribución (DFL).
Propuesto en Pérdida Focal Generalizada https://ieeexplore.ieee.org/document/9792391
Código fuente en ultralytics/nn/modules/block.py
__init__(c1=16)
Inicializa una capa convolucional con un número determinado de canales de entrada.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
Aplica una capa transformadora en la entrada tensor 'x' y devuelve un tensor.
ultralytics.nn.modules.block.Proto
Bases: Module
YOLOv8 máscara Módulo proto para modelos de segmentación.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c_=256, c2=32)
Inicializa el módulo YOLOv8 mask Proto con el número especificado de protos y máscaras.
Los argumentos de entrada son ch_in, número de protos, número de máscaras.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
Realiza un pase hacia delante a través de capas utilizando una imagen de entrada sobremuestreada.
ultralytics.nn.modules.block.HGStem
Bases: Module
StemBlock de PPHGNetV2 con 5 convoluciones y un maxpool2d.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, cm, c2)
Inicializa la capa SPP con los canales de entrada/salida y los tamaños de núcleo especificados para la agrupación máxima.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
Paso adelante de una capa troncal PPHGNetV2.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.HGBlock
Bases: Module
HG_Bloque de PPHGNetV2 con 2 convoluciones y LightConv.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, cm, c2, k=3, n=6, lightconv=False, shortcut=False, act=nn.ReLU())
Inicializa un Cuello de Botella CSP con 1 convolución utilizando los canales de entrada y salida especificados.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
Paso adelante de una capa troncal PPHGNetV2.
ultralytics.nn.modules.block.SPP
Bases: Module
Capa de Agrupación de Pirámides Espaciales (SPP) https://arxiv.org/abs/1406.4729.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, k=(5, 9, 13))
Inicializa la capa SPP con los canales de entrada/salida y los tamaños de los núcleos de agrupación.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
Paso adelante de la capa SPP, realizando la agrupación piramidal espacial.
ultralytics.nn.modules.block.SPPF
Bases: Module
Capa de Agrupación de Pirámides Espaciales - Rápida (SPPF) para YOLOv5 por Glenn Jocher.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, k=5)
Inicializa la capa SPPF con los canales de entrada/salida y el tamaño del núcleo dados.
Este módulo es equivalente a SPP(k=(5, 9, 13)).
Código fuente en ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.C1
Bases: Module
CSP Cuello de botella con 1 convolución.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1)
Inicializa el Cuello de Botella CSP con configuraciones para 1 convolución con argumentos ch_in, ch_out, número.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2
Bases: Module
CSP Cuello de botella con 2 convoluciones.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inicializa el módulo CSP Cuello de Botella con 2 convoluciones con los argumentos ch_in, ch_out, número, atajo, grupos, expansión.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.C2f
Bases: Module
Aplicación más rápida del cuello de botella CSP con 2 convoluciones.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=False, g=1, e=0.5)
Inicializa la capa de cuello de botella CSP con dos convoluciones con argumentos ch_in, ch_out, número, atajo, grupos, expansión.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
Paso adelante utilizando split() en lugar de chunk().
ultralytics.nn.modules.block.C3
Bases: Module
CSP Cuello de botella con 3 convoluciones.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inicializa el Cuello de Botella CSP con los valores dados de canales, número, acceso directo, grupos y expansión.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3x
Bases: C3
Módulo C3 con circunvoluciones cruzadas.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inicializa la instancia C3TR y establece los parámetros por defecto.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepC3
Bases: Module
Rep C3.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=3, e=1.0)
Inicializa el CSP Cuello de Botella con una sola convolución utilizando los canales de entrada, los canales de salida y el número.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3TR
Bases: C3
Módulo C3 con TransformerBlock().
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inicializa el módulo C3Ghost con GhostBottleneck().
ultralytics.nn.modules.block.C3Ghost
Bases: C3
Módulo C3 con GhostBottleneck().
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inicializa el módulo "SPP" con varios tamaños de agrupación para la agrupación piramidal espacial.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.GhostBottleneck
Bases: Module
Cuello de botella fantasma https://github.com/huawei-noah/ghostnet.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, k=3, s=1)
Inicializa el módulo GhostBottleneck con los argumentos ch_in, ch_out, kernel, stride.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.Bottleneck
Bases: Module
Cuello de botella estándar.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Inicializa un módulo cuello de botella con los canales de entrada/salida, opción de acceso directo, grupo, núcleos y expansión.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.BottleneckCSP
Bases: Module
Cuello de botella CSP https://github.com/WongKinYiu/CrossStagePartialNetworks.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inicializa el Cuello de Botella CSP dados los argumentos para ch_in, ch_out, número, atajo, grupos, expansión.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
Aplica un cuello de botella CSP con 3 convoluciones.
ultralytics.nn.modules.block.ResNetBlock
Bases: Module
Bloque ResNet con capas de convolución estándar.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, e=4)
Inicializa la convolución con los parámetros dados.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ResNetLayer
Bases: Module
Capa ResNet con varios bloques ResNet.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, is_first=False, n=1, e=4)
Inicializa el ResNetLayer dados los argumentos.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.MaxSigmoidAttnBlock
Bases: Module
Bloqueo máximo de atención sigmoidea.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, nh=1, ec=128, gc=512, scale=False)
Inicializa MaxSigmoidAttnBlock con los argumentos especificados.
Código fuente en ultralytics/nn/modules/block.py
forward(x, guide)
Proceso de avance.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2fAttn
Bases: Module
Módulo C2f con un módulo attn adicional.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, ec=128, nh=1, gc=512, shortcut=False, g=1, e=0.5)
Inicializa la capa de cuello de botella CSP con dos convoluciones con argumentos ch_in, ch_out, número, atajo, grupos, expansión.
Código fuente en ultralytics/nn/modules/block.py
forward(x, guide)
Paso hacia delante a través de la capa C2f.
forward_split(x, guide)
Paso adelante utilizando split() en lugar de chunk().
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ImagePoolingAttn
Bases: Module
ImagePoolingAttn: Mejora las incrustaciones de texto con información que tenga en cuenta las imágenes.
Código fuente en ultralytics/nn/modules/block.py
__init__(ec=256, ch=(), ct=512, nh=8, k=3, scale=False)
Inicializa ImagePoolingAttn con los argumentos especificados.
Código fuente en ultralytics/nn/modules/block.py
forward(x, text)
Ejecuta el mecanismo de atención sobre la entrada tensor x y la guía tensor.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ContrastiveHead
Bases: Module
Contrastive Head para YOLO-World calcula las puntuaciones región-texto según la similitud entre imagen y texto texto.
Código fuente en ultralytics/nn/modules/block.py
__init__()
Inicializa ContrastiveHead con los parámetros de similitud región-texto especificados.
Código fuente en ultralytics/nn/modules/block.py
forward(x, w)
Función de avance del aprendizaje contrastivo.
ultralytics.nn.modules.block.BNContrastiveHead
Bases: Module
Cabeza contrastiva de norma por lotes para YOLO-Mundo utilizando la norma por lotes en lugar de la normalización l2.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
embed_dims |
int
|
Incrusta las dimensiones de los elementos de texto e imagen. |
necesario |
Código fuente en ultralytics/nn/modules/block.py
__init__(embed_dims)
Inicializa ContrastiveHead con los parámetros de similitud región-texto.
Código fuente en ultralytics/nn/modules/block.py
forward(x, w)
Función de avance del aprendizaje contrastivo.
ultralytics.nn.modules.block.RepBottleneck
Bases: Bottleneck
Rep cuello de botella.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Inicializa un módulo RepBottleneck con canales de entrada/salida personalizables, opción de acceso directo, grupos y ratio de expansión. relación.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepCSP
Bases: C3
Rep CSP Cuello de botella con 3 convoluciones.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Inicializa la capa RepCSP con los canales, repeticiones, atajos, grupos y ratio de expansión dados.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepNCSPELAN4
Bases: Module
CSP-ELAN.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, c3, c4, n=1)
Inicializa la capa CSP-ELAN con los tamaños de canal, repeticiones y convoluciones especificados.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
Paso adelante a través de la capa RepNCSPELAN4.
forward_split(x)
Paso adelante utilizando split() en lugar de chunk().
ultralytics.nn.modules.block.ADown
Bases: Module
ADown.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2)
Inicializa el módulo ADown con capas de convolución para reducir la muestra de entrada de los canales c1 a c2.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
Paso hacia delante a través de la capa ADown.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.SPPELAN
Bases: Module
SPP-ELAN.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2, c3, k=5)
Inicializa el bloque SPP-ELAN con las capas de convolución y agrupación máxima para la agrupación de pirámides espaciales.
Código fuente en ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.Silence
Bases: Module
Silencio.
Código fuente en ultralytics/nn/modules/block.py
__init__()
ultralytics.nn.modules.block.CBLinear
Bases: Module
CBLineal.
Código fuente en ultralytics/nn/modules/block.py
__init__(c1, c2s, k=1, s=1, p=None, g=1)
Inicializa el módulo CBLinear, pasando las entradas sin cambios.
Código fuente en ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.CBFuse
Bases: Module
CBFuse.
Código fuente en ultralytics/nn/modules/block.py
__init__(idx)
Inicializa el módulo CBFuse con el índice de capas para la fusión selectiva de rasgos.
forward(xs)
Paso hacia delante a través de la capa CBFuse.