Ссылка для ultralytics/nn/modules/block.py
Примечание
Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/nn/modules/block .py. Если ты заметил проблему, пожалуйста, помоги исправить ее, отправив Pull Request 🛠️. Спасибо 🙏!
ultralytics.nn.modules.block.DFL
Базы: Module
Интегральный модуль распределения фокусных потерь (DFL).
Предложено в генерализованной очаговой потере https://ieeexplore.ieee.org/document/9792391
Исходный код в ultralytics/nn/modules/block.py
__init__(c1=16)
Инициализируй конволюционный слой с заданным количеством входных каналов.
Исходный код в ultralytics/nn/modules/block.py
forward(x)
Примени слой трансформации к входному tensor 'x' и верни tensor.
ultralytics.nn.modules.block.Proto
Базы: Module
YOLOv8 Модуль mask Proto для моделей сегментации.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c_=256, c2=32)
Инициализирует модуль YOLOv8 mask Proto с указанным количеством протосов и масок.
Входные аргументы: ch_in, количество протосов, количество масок.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.HGStem
Базы: Module
StemBlock из PPHGNetV2 с 5 свертками и одним maxpool2d.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, cm, c2)
Инициализируй SPP-слой с входными/выходными каналами и заданными размерами ядра для максимального пулинга.
Исходный код в ultralytics/nn/modules/block.py
forward(x)
Проход вперед магистрального уровня PPHGNetV2.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.HGBlock
Базы: Module
HG_Block из PPHGNetV2 с 2 свертками и LightConv.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, cm, c2, k=3, n=6, lightconv=False, shortcut=False, act=nn.ReLU())
Инициализирует CSP Bottleneck с 1 сверткой, используя указанные входные и выходные каналы.
Исходный код в ultralytics/nn/modules/block.py
forward(x)
Проход вперед магистрального уровня PPHGNetV2.
ultralytics.nn.modules.block.SPP
Базы: Module
Слой Spatial Pyramid Pooling (SPP) https://arxiv.org/abs/1406.4729.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, k=(5, 9, 13))
Инициализируй SPP-слой с каналами ввода/вывода и размерами ядра пулинга.
Исходный код в ultralytics/nn/modules/block.py
forward(x)
Проход вперед слоя SPP, выполняющего пространственное пирамидное объединение.
ultralytics.nn.modules.block.SPPF
Базы: Module
Слой Spatial Pyramid Pooling - Fast (SPPF) для YOLOv5 от Глена Джочера.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, k=5)
Инициализирует слой SPPF с заданными входными/выходными каналами и размером ядра.
Этот модуль эквивалентен SPP(k=(5, 9, 13)).
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C1
Базы: Module
CSP Bottleneck с 1 сверткой.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1)
Инициализирует CSP Bottleneck с конфигурациями для 1 свертки с аргументами ch_in, ch_out, number.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2
Базы: Module
CSP Bottleneck с двумя свертками.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Инициализирует модуль CSP Bottleneck with 2 convolutions с аргументами ch_in, ch_out, number, shortcut, группы, расширение.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2f
Базы: Module
Более быстрая реализация CSP Bottleneck с помощью двух сверток.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=False, g=1, e=0.5)
Инициализируй слой CSP bottleneck с двумя свертками с аргументами ch_in, ch_out, number, shortcut, groups, расширение.
Исходный код в ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
Пройдите вперед, используя split() вместо chunk().
ultralytics.nn.modules.block.C3
Базы: Module
CSP Bottleneck с 3 свертками.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Инициализируй CSP Bottleneck с заданными значениями каналов, количества, ярлыка, групп и расширения.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3x
Базы: C3
Модуль C3 с перекрестными конвульсиями.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Инициализируй экземпляр C3TR и установи параметры по умолчанию.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepC3
Базы: Module
Rep C3.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=3, e=1.0)
Инициализируй CSP Bottleneck с помощью одной свертки, используя входные каналы, выходные каналы и число.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3TR
Базы: C3
Модуль C3 с функцией TransformerBlock().
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Инициализируй модуль C3Ghost с помощью функции GhostBottleneck().
ultralytics.nn.modules.block.C3Ghost
Базы: C3
Модуль C3 с функцией GhostBottleneck().
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Инициализируй модуль 'SPP' с различными размерами пула для пула пространственных пирамид.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.GhostBottleneck
Базы: Module
Призрачная бутылка https://github.com/huawei-noah/ghostnet.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, k=3, s=1)
Инициализирует модуль GhostBottleneck с аргументами ch_in, ch_out, kernel, stride.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.Bottleneck
Базы: Module
Стандартное узкое место.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Инициализирует модуль узкого места с заданными каналами ввода/вывода, опцией быстрого доступа, группой, ядрами и расширением.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.BottleneckCSP
Базы: Module
CSP Bottleneck https://github.com/WongKinYiu/CrossStagePartialNetworks.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Инициализирует CSP Bottleneck, задав аргументы для ch_in, ch_out, number, shortcut, groups, expansion.
Исходный код в ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.ResNetBlock
Базы: Module
Блок ResNet со стандартными слоями свертки.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, e=4)
Инициализируй свертку с заданными параметрами.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ResNetLayer
Базы: Module
Слой ResNet с несколькими блоками ResNet.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, is_first=False, n=1, e=4)
Инициализирует ResNetLayer, заданный аргументами.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.MaxSigmoidAttnBlock
Базы: Module
Максимальный сигмовидный блок внимания.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, nh=1, ec=128, gc=512, scale=False)
Инициализирует MaxSigmoidAttnBlock с указанными аргументами.
Исходный код в ultralytics/nn/modules/block.py
forward(x, guide)
Форвардный процесс.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2fAttn
Базы: Module
Модуль C2f с дополнительным модулем attn.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, ec=128, nh=1, gc=512, shortcut=False, g=1, e=0.5)
Инициализируй слой CSP bottleneck с двумя свертками с аргументами ch_in, ch_out, number, shortcut, groups, расширение.
Исходный код в ultralytics/nn/modules/block.py
forward(x, guide)
Пройди вперед через слой C2f.
forward_split(x, guide)
Пройдите вперед, используя split() вместо chunk().
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ImagePoolingAttn
Базы: Module
ImagePoolingAttn: улучшай текстовые вкрапления с помощью информации об изображениях.
Исходный код в ultralytics/nn/modules/block.py
__init__(ec=256, ch=(), ct=512, nh=8, k=3, scale=False)
Инициализирует ImagePoolingAttn с указанными аргументами.
Исходный код в ultralytics/nn/modules/block.py
forward(x, text)
Выполняет механизм внимания на входе tensor x и проводнике tensor.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ContrastiveHead
Базы: Module
Контрастная голова для YOLO-World вычисляет оценки "регион-текст" в соответствии со сходством между изображением и текстом. особенности.
Исходный код в ultralytics/nn/modules/block.py
__init__()
Инициализирует ContrastiveHead с заданными параметрами сходства региона и текста.
Исходный код в ultralytics/nn/modules/block.py
forward(x, w)
Передовая функция контрастного обучения.
ultralytics.nn.modules.block.BNContrastiveHead
Базы: Module
Контрастная голова для YOLO-World, использующая пакетную норму вместо l2-нормировки.
Параметры:
Имя | Тип | Описание | По умолчанию |
---|---|---|---|
embed_dims |
int
|
Вложи размеры в особенности текста и изображения. |
требуется |
Исходный код в ultralytics/nn/modules/block.py
__init__(embed_dims)
Инициализируй ContrastiveHead с параметрами сходства региона и текста.
Исходный код в ultralytics/nn/modules/block.py
forward(x, w)
Передовая функция контрастного обучения.
ultralytics.nn.modules.block.RepBottleneck
Базы: Bottleneck
Реп узкое место.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Инициализирует модуль RepBottleneck с настраиваемыми каналами входа/выхода, опцией быстрого доступа, группами и коэффициентом расширения Соотношение.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepCSP
Базы: C3
Rep CSP Bottleneck с 3 свертками.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Инициализирует слой RepCSP с заданными каналами, повторами, ярлыками, группами и коэффициентом расширения.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepNCSPELAN4
Базы: Module
CSP-ELAN.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, c3, c4, n=1)
Инициализирует слой CSP-ELAN с заданными размерами каналов, повторениями и свертками.
Исходный код в ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
Пройдите вперед, используя split() вместо chunk().
ultralytics.nn.modules.block.ADown
Базы: Module
ADown.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2)
Инициализирует модуль ADown со слоями свертки для понижения дискретизации входного сигнала из каналов c1 в c2.
Исходный код в ultralytics/nn/modules/block.py
forward(x)
Проход вперед через слой ADown.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.SPPELAN
Базы: Module
SPP-ELAN.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2, c3, k=5)
Инициализирует блок SPP-ELAN со слоями свертки и максимального пулинга для пулинга пространственной пирамиды.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.Silence
Базы: Module
Молчание.
Исходный код в ultralytics/nn/modules/block.py
__init__()
ultralytics.nn.modules.block.CBLinear
Базы: Module
CBLinear.
Исходный код в ultralytics/nn/modules/block.py
__init__(c1, c2s, k=1, s=1, p=None, g=1)
Инициализирует модуль CBLinear, передавая входные данные без изменений.
Исходный код в ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.CBFuse
Базы: Module
CBFuse.
Исходный код в ultralytics/nn/modules/block.py
__init__(idx)
forward(xs)
Проход вперед через слой CBFuse.