timm documentation

ハイブリッドビジョントランスフォーマー

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

ハイブリッドビジョントランスフォーマー

カテゴリ: ハイブリッド · コード: timm/models/vision_transformer_hybrid.py · アーキテクチャ: 14 · 事前学習済み重み: 14 (ImageNet-1kの結果付き: 7)

線形パッチ埋め込みではなく、畳み込みバックボーン(ResNet / ResNetV2のステムまたはステージ)からパッチトークンを取得するVision Transformers。この場合、VisionTransformerクラスを使用します。

論文

使用方法

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

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

モデルAPIのサポート

機能VisionTransformer
features_only=True 特徴マップ✓ (strides 32, 32, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
入力サイズは任意で、変更は不要です✗
set_input_size()✓
dynamic_img_size=True✓
set_grad_checkpointing()✓
レイヤーごとのLR減衰(group_matcher())✓
reset_classifier()✓

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

アーキテクチャ

アーキテクチャパラメータ数 (M)GMACs入力サイズ事前学習済み重み
vit_base_mci_22486.3517.732243
vit_base_r26_s32_224101.386.812240
vit_base_r50_s16_22497.8921.662241
vit_base_r50_s16_38498.9567.433841
vit_base_resnet26d_224101.406.972240
vit_base_resnet50d_224110.978.732240
vit_large_r50_s32_224328.9919.582242
vit_large_r50_s32_384329.0957.433841
vit_small_r26_s32_22436.433.562242
vit_small_r26_s32_38436.4710.433841
vit_small_resnet26d_22463.615.072240
vit_small_resnet50d_s16_22457.5313.482240
vit_tiny_r_s16_p8_2246.340.442242
vit_tiny_r_s16_p8_3846.361.343841

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

事前学習済み重み

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

重みはソースごとにグループ化されます。ソースは、事前学習設定で指定されている論文または出典、それ以外の場合はtimm以外の Hub 組織、さらにそれ以外の場合は事前学習タグの最初の部分(architecture.tag)です。重みが 1 つしかないソースは、other にリストされます。

augreg タグ

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
vit_large_r50_s32_224.augreg_in21k224--21843apache-2.0timm
vit_large_r50_s32_224.augreg_in21k_ft_in1k22484.4342241000apache-2.0timm
vit_large_r50_s32_384.augreg_in21k_ft_in1k38486.1883841000apache-2.0timm
vit_small_r26_s32_224.augreg_in21k224--21843apache-2.0timm
vit_small_r26_s32_224.augreg_in21k_ft_in1k22481.8562241000apache-2.0timm
vit_small_r26_s32_384.augreg_in21k_ft_in1k38484.0663841000apache-2.0timm
vit_tiny_r_s16_p8_224.augreg_in21k224--21843apache-2.0timm
vit_tiny_r_s16_p8_224.augreg_in21k_ft_in1k22471.8002241000apache-2.0timm
vit_tiny_r_s16_p8_384.augreg_in21k_ft_in1k38475.9623841000apache-2.0timm

元のタグ

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
vit_base_r50_s16_224.orig_in21k224--なしapache-2.0timm
vit_base_r50_s16_384.orig_in21k_ft_in1k38484.9883841000apache-2.0timm

apple

Hugging Face Hub上でappleがホストしています。

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
vit_base_mci_224.apple_mclip224--512apple-amlrapple/mobileclip_b_timm
vit_base_mci_224.apple_mclip_lt224--512apple-amlrapple/mobileclip_b_lt_timm

その他

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
vit_base_mci_224.apple_mclip2_dfndr2b224--512apple-amlrtimm
Update on GitHub