timm documentation

FocalNet

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.30).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

FocalNet

カテゴリ: ハイブリッド · コード: timm/models/focalnet.py · アーキテクチャ: 12 · 事前学習済み重み: 12 (ImageNet-1kの結果を含む6つ)

Focal Modulation Networks (FocalNet) は、セルフアテンションをフォーカルモジュレーションに置き換えます。フォーカルモジュレーションは、ディプスワイズ畳み込みによって複数の粒度でコンテキストを集約し、各クエリトークンにゲートします。

論文

使用方法

事前学習済み重みの読み込み、画像の前処理、ファインチューニングについてはクイックスタートを、モデルをバックボーンとして使用する方法については特徴量抽出を参照してください。

>>> import timm
>>> model = timm.create_model('focalnet_tiny_srf.ms_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

モデルAPIのサポート

機能FocalNet
features_only=True 特徴マップ✓ (ストライド 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
入力サイズは任意で、変更は不要です✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
レイヤーごとのLR減衰(group_matcher())✓
reset_classifier()✓

✓ 対応、✓ (square) 正方形サイズのみ対応、✗ 非対応、– 不要(モデルは任意の入力サイズを既に受け付ける)。ランダム初期化されたモデル(FocalNet の focalnet_tiny_srf、224)で各機能を実行して確認しています。同じクラスの他のアーキテクチャも同様に動作すると予想されます。詳細はModel APIページを参照してください。

アーキテクチャ

アーキテクチャパラメータ数 (M)GMACs入力サイズ事前学習済み重み
focalnet_base_lrf88.7515.432241
focalnet_base_srf88.1515.282241
focalnet_huge_fl3745.28118.262241
focalnet_huge_fl4686.46118.902241
focalnet_large_fl3239.13105.063841
focalnet_large_fl4239.32105.203841
focalnet_small_lrf50.348.742241
focalnet_small_srf49.898.622241
focalnet_tiny_lrf28.654.492241
focalnet_tiny_srf28.434.422241
focalnet_xlarge_fl3408.79185.613841
focalnet_xlarge_fl4409.03185.793841

GMACsとパラメータ数は、各アーキテクチャのデフォルトの入力サイズと分類器に基づいており、ベンチマーク結果から得られたものです。

事前学習済み重み

Top-1 は、results-imagenet.csv に記載された評価サイズでの ImageNet-1k 検証精度です。- は、ImageNet-1k の結果がない重み(例: プリトレーニング用または画像-テキスト用重み)を示します。Classes は事前学習済み分類器のサイズで、分類器を持たない重みについては none です。

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
focalnet_base_lrf.ms_in1k22483.8282241000mittimm
focalnet_base_srf.ms_in1k22483.8082241000mittimm
focalnet_huge_fl3.ms_in22k224--21842mittimm
focalnet_huge_fl4.ms_in22k224--なしmittimm
focalnet_large_fl3.ms_in22k384--21842mittimm
focalnet_large_fl4.ms_in22k384--21842mittimm
focalnet_small_lrf.ms_in1k22483.5082241000mittimm
focalnet_small_srf.ms_in1k22483.4282241000mittimm
focalnet_tiny_lrf.ms_in1k22482.1742241000mittimm
focalnet_tiny_srf.ms_in1k22482.1462241000mittimm
focalnet_xlarge_fl3.ms_in22k384--21842mittimm
focalnet_xlarge_fl4.ms_in22k384--21842mittimm
Update on GitHub