参考 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 セグメンテーションモデル用のマスクプロトモジュール。
ソースコード ultralytics/nn/modules/block.py
__init__(c1, c_=256, c2=32)
YOLOv8 マスク Proto モジュールを、指定された数のプロトとマスクで初期化する。
入力引数はch_in、プロトの数、マスクの数。
ソースコード ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.HGStem
ベース: Module
5つの畳み込みと1つのmaxpool2dを持つPPHGNetV2のステムブロック。
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
2つのコンボリューションとLightConvを持つPPHGNetV2のHG_ブロック。
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())
指定された入出力チャネルを使用する 1 畳み込みで CSP ボトルネックを初期化します。
ソースコード ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.SPP
ベース: Module
SPP(Spatial Pyramid Pooling)レイヤーhttps://arxiv.org/abs/1406.4729。
ソースコード ultralytics/nn/modules/block.py
__init__(c1, c2, k=(5, 9, 13))
入出力チャンネルとプーリングカーネルサイズでSPPレイヤーを初期化する。
ソースコード ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.SPPF
ベース: Module
Spatial Pyramid Pooling - Fast (SPPF) layer forYOLOv5 by Glenn Jocher.
ソースコード 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
1回の畳み込みによるCSPのボトルネック。
ソースコード ultralytics/nn/modules/block.py
__init__(c1, c2, n=1)
引数 ch_in, ch_out, number を持つ 1 畳み込みのコンフィギュレーションで CSP Bottleneck を初期化します。
ソースコード ultralytics/nn/modules/block.py
ultralytics.nn.modules.block.C2
ベース: Module
2つの畳み込みによるCSPボトルネック。
ソースコード 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
2つの畳み込みによるCSPボトルネックの高速化。
ソースコード ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=False, g=1, e=0.5)
引数 ch_in, ch_out, number, shortcut, groups を持つ2つの畳み込みでCSPボトルネック層を初期化する、 展開。
ソースコード ultralytics/nn/modules/block.py
forward(x)
forward_split(x)
ultralytics.nn.modules.block.C3
ベース: Module
3つの畳み込みによるCSPボトルネック。
ソースコード ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
指定されたチャネル、数、ショートカット、グループ、および拡張値で CSP ボトルネックを初期化します。
ソースコード 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
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)
GhostBottleneck()でC3Ghostモジュールを初期化する。
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)
引数 ch_in, ch_out, kernel, stride で GhostBottleneck モジュールを初期化する。
ソースコード 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 ボトルネック https://github.com/WongKinYiu/CrossStagePartialNetworks.
ソースコード ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, shortcut=True, g=1, e=0.5)
ch_in, ch_out, number, shortcut, groups, expansion の引数が指定された場合、CSP Bottleneck を初期化します。
ソースコード ultralytics/nn/modules/block.py
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
Attnモジュールを追加したC2fモジュール。
ソースコード ultralytics/nn/modules/block.py
__init__(c1, c2, n=1, ec=128, nh=1, gc=512, shortcut=False, g=1, e=0.5)
引数 ch_in, ch_out, number, shortcut, groups を持つ2つの畳み込みでCSPボトルネック層を初期化する、 展開。
ソースコード ultralytics/nn/modules/block.py
forward(x, guide)
forward_split(x, guide)
chunk() の代わりに split() を使用した前方パス。
ソースコード 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
Contrastive Head forYOLO-World 画像とテキストの特徴量の類似度に応じて、領域とテキストのスコアを計算する。 を計算する。
ソースコード ultralytics/nn/modules/block.py
__init__()
ContrastiveHeadを、指定されたリージョン・テキスト類似度パラメータで初期化する。
ソースコード ultralytics/nn/modules/block.py
forward(x, w)
ultralytics.nn.modules.block.BNContrastiveHead
ベース: Module
YOLO-L2 正規化の代わりにバッチノルムを使用する世界のためのバッチノルム対照ヘッド。
パラメーター
名称 | タイプ | 説明 | デフォルト |
---|---|---|---|
embed_dims |
int
|
テキストと画像の特徴を埋め込む。 |
必須 |
ソースコード ultralytics/nn/modules/block.py
__init__(embed_dims)
ContrastiveHead を地域-テキスト類似度パラメータで初期化する。
ソースコード ultralytics/nn/modules/block.py
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
3つのコンボリューションを持つレップCSPボトルネック。
ソースコード 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)
chunk() の代わりに split() を使用した前方パス。
ultralytics.nn.modules.block.ADown
ベース: Module
ADown。
ソースコード ultralytics/nn/modules/block.py
__init__(c1, c2)
チャンネル c1 から c2 の入力をダウンサンプリングする畳み込みレイヤーを持つ ADown モジュールを初期化する。
ソースコード ultralytics/nn/modules/block.py
forward(x)
ADダウンレイヤーをフォワードパス。
ソースコード 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レイヤーを通過するフォワードパス。