Tài liệu tham khảo cho ultralytics/nn/modules/block.py
Ghi
Tệp này có sẵn tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/nn/mô-đun/block.py. Nếu bạn phát hiện ra một vấn đề, vui lòng giúp khắc phục nó bằng cách đóng góp Yêu cầu 🛠️ kéo. Cảm ơn bạn 🙏 !
ultralytics.nn.modules.block.DFL
Căn cứ: Module
Mô-đun tích hợp của tổn thất đầu mối phân phối (DFL).
Đề xuất trong https://ieeexplore.ieee.org/document/9792391 mất khu trú tổng quát
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1=16)
Khởi tạo một lớp tích chập với một số kênh đầu vào nhất định.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
Áp dụng một lớp biến áp trên đầu vào tensor 'x' và trả về a tensor.
ultralytics.nn.modules.block.Proto
Căn cứ: Module
YOLOv8 mô-đun mặt nạ Proto cho các mô hình phân khúc.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c_=256, c2=32)
Khởi tạo các YOLOv8 mô-đun mặt nạ Proto với số lượng protos và mặt nạ được chỉ định.
Đối số đầu vào là ch_in, số lượng proto, số mặt nạ.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.HGStem
Căn cứ: Module
StemBlock của PPHGNetV2 với 5 kết cấu và một maxpool2d.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, cm, c2)
Khởi tạo lớp SPP với các kênh đầu vào / đầu ra và kích thước hạt nhân được chỉ định để gộp tối đa.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
Chuyển tiếp vượt qua lớp xương sống PPHGNetV2.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.HGBlock
Căn cứ: Module
HG_Block của PPHGNetV2 với 2 kết cấu và LightConv.
https://github.com/PaddlePaddle/PaddleDetection/blob/develop/ppdet/modeling/backbones/hgnet_v2.py
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, cm, c2, k=3, n=6, lightconv=False, shortcut=False, act=nn.ReLU())
Khởi tạo CSP Bottleneck với 1 tích chập sử dụng các kênh đầu vào và đầu ra được chỉ định.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
Chuyển tiếp vượt qua lớp xương sống PPHGNetV2.
ultralytics.nn.modules.block.SPP
Căn cứ: Module
Lớp Spatial Pyramid Pooling (SPP) https://arxiv.org/abs/1406.4729.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, k=(5, 9, 13))
Khởi tạo lớp SPP với các kênh đầu vào / đầu ra và tổng hợp kích thước hạt nhân.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
Chuyển tiếp vượt qua lớp SPP, thực hiện gộp kim tự tháp không gian.
ultralytics.nn.modules.block.SPPF
Căn cứ: Module
Spatial Pyramid Pooling - Lớp nhanh (SPPF) cho YOLOv5 bởi Glenn Jocher.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, k=5)
Khởi tạo lớp SPPF với các kênh đầu vào / đầu ra và kích thước hạt nhân nhất định.
Mô-đun này tương đương với SPP (k = (5, 9, 13)).
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C1
Căn cứ: Module
Nút cổ chai CSP với 1 tích chập.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1)
Khởi tạo CSP Bottleneck với các cấu hình cho 1 tích chập với các đối số ch_in, ch_out, number.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2
Căn cứ: Module
Nút cổ chai CSP với 2 kết cấu.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Khởi tạo CSP Bottleneck với 2 module convolutions với các đối số ch_in, ch_out, number, shortcut, nhóm, mở rộng.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2f
Căn cứ: Module
Thực hiện nhanh hơn nút thắt cổ chai CSP với 2 kết cấu.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=False, g=1, e=0.5)
Khởi tạo lớp nút cổ chai CSP với hai tích chập với các đối số ch_in, ch_out, số, phím tắt, nhóm, Mở rộng.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
Chuyển tiếp vượt qua bằng cách sử dụng split() thay vì chunk().
ultralytics.nn.modules.block.C3
Căn cứ: Module
Nút cổ chai CSP với 3 kết cấu.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Khởi tạo CSP Bottleneck với các kênh, số, phím tắt, nhóm và giá trị mở rộng đã cho.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3x
Căn cứ: C3
Mô-đun C3 với các kết cấu chéo.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Khởi tạo phiên bản C3TR và đặt tham số mặc định.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepC3
Căn cứ: Module
Đại diện C3.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=3, e=1.0)
Khởi tạo CSP Bottleneck với một tích chập duy nhất bằng cách sử dụng các kênh đầu vào, kênh đầu ra và số.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C3TR
Căn cứ: C3
Mô-đun C3 với TransformerBlock ().
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Khởi tạo mô-đun C3Ghost với GhostBottleneck().
ultralytics.nn.modules.block.C3Ghost
Căn cứ: C3
Mô-đun C3 với GhostBottleneck().
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Khởi tạo mô-đun 'SPP' với các kích thước gộp khác nhau để gộp kim tự tháp không gian.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.GhostBottleneck
Căn cứ: Module
Nút cổ chai ma https://github.com/huawei-noah/ghostnet.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, k=3, s=1)
Khởi tạo mô-đun GhostBottleneck với các đối số ch_in, ch_out, kernel, stride.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.Bottleneck
Căn cứ: Module
Nút thắt cổ chai tiêu chuẩn.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Khởi tạo mô-đun nút cổ chai với các kênh đầu vào / đầu ra nhất định, tùy chọn phím tắt, nhóm, hạt nhân và Mở rộng.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.BottleneckCSP
Căn cứ: Module
Nút thắt cổ chai CSP https://github.com/WongKinYiu/CrossStagePartialNetworks.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Khởi tạo nút cổ chai CSP cho các đối số cho ch_in, ch_out, số, phím tắt, nhóm, mở rộng.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
ultralytics.nn.modules.block.ResNetBlock
Căn cứ: Module
Khối ResNet với các lớp tích chập tiêu chuẩn.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, e=4)
Khởi tạo tích chập với các tham số đã cho.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ResNetLayer
Căn cứ: Module
Lớp ResNet với nhiều khối ResNet.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, s=1, is_first=False, n=1, e=4)
Khởi tạo các đối số đã cho của ResNetLayer.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.MaxSigmoidAttnBlock
Căn cứ: Module
Khối chú ý tối đa Sigma.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, nh=1, ec=128, gc=512, scale=False)
Khởi tạo MaxSigmoidAttnBlock với các đối số được chỉ định.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x, guide)
Quá trình chuyển tiếp.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2fAttn
Căn cứ: Module
Mô-đun C2f với một mô-đun attn bổ sung.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, ec=128, nh=1, gc=512, shortcut=False, g=1, e=0.5)
Khởi tạo lớp nút cổ chai CSP với hai tích chập với các đối số ch_in, ch_out, số, phím tắt, nhóm, Mở rộng.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x, guide)
Chuyển tiếp đi qua lớp C2f.
forward_split(x, guide)
Chuyển tiếp vượt qua bằng cách sử dụng split() thay vì chunk().
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ImagePoolingAttn
Căn cứ: Module
ImagePoolingAttn: Tăng cường nhúng văn bản với thông tin nhận biết hình ảnh.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(ec=256, ch=(), ct=512, nh=8, k=3, scale=False)
Khởi tạo ImagePoolingAttn với các đối số được chỉ định.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x, text)
Thực hiện cơ chế chú ý trên đầu vào tensor x và hướng dẫn tensor.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.ContrastiveHead
Căn cứ: Module
Đầu tương phản cho YOLO-Thế giới tính điểm văn bản khu vực theo sự giống nhau giữa hình ảnh và văn bản Tính năng.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__()
Khởi tạo ContrastiveHead với các tham số tương tự vùng-văn bản được chỉ định.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x, w)
Chức năng chuyển tiếp của học tập tương phản.
ultralytics.nn.modules.block.BNContrastiveHead
Căn cứ: Module
Batch Norm Contrastive Head cho YOLO-Thế giới sử dụng định mức hàng loạt thay vì chuẩn hóa l2.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
embed_dims |
int
|
Nhúng kích thước của các tính năng văn bản và hình ảnh. |
bắt buộc |
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(embed_dims)
Khởi tạo ContrastiveHead với các tham số tương tự vùng-văn bản.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x, w)
Chức năng chuyển tiếp của học tập tương phản.
ultralytics.nn.modules.block.RepBottleneck
Căn cứ: Bottleneck
Nút cổ chai đại diện.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, shortcut=True, g=1, k=(3, 3), e=0.5)
Khởi tạo mô-đun RepBottleneck với các kênh vào / ra có thể tùy chỉnh, tùy chọn phím tắt, nhóm và mở rộng tỉ lệ.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepCSP
Căn cứ: C3
Rep CSP Bottleneck với 3 kết cấu.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
Khởi tạo lớp RepCSP với các kênh, lặp lại, phím tắt, nhóm và tỷ lệ mở rộng nhất định.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.RepNCSPELAN4
Căn cứ: Module
CSP-ELAN.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, c3, c4, n=1)
Khởi tạo lớp CSP-ELAN với kích thước kênh, lặp lại và kết cấu được chỉ định.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
Chuyển tiếp vượt qua bằng cách sử dụng split() thay vì chunk().
ultralytics.nn.modules.block.ADown
Căn cứ: Module
Quảng cáo.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2)
Khởi tạo mô-đun ADown với các lớp tích chập để downsample input từ các kênh c1 đến c2.
Mã nguồn trong ultralytics/nn/modules/block.py
forward(x)
Chuyển tiếp đi qua lớp ADown.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.SPPELAN
Căn cứ: Module
SPP-ELAN.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2, c3, k=5)
Khởi tạo khối SPP-ELAN với các lớp tích chập và tổng hợp tối đa để gộp kim tự tháp không gian.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.Silence
Căn cứ: Module
Sự yên lặng.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__()
ultralytics.nn.modules.block.CBLinear
Căn cứ: Module
CBLinear.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(c1, c2s, k=1, s=1, p=None, g=1)
Khởi tạo mô-đun CBLinear, truyền đầu vào không thay đổi.
Mã nguồn trong ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.CBFuse
Căn cứ: Module
CBFuse.
Mã nguồn trong ultralytics/nn/modules/block.py
__init__(idx)
forward(xs)
Chuyển tiếp đi qua lớp CBFuse.