timm documentation

XCiT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

XCiT

カテゴリ: ViT · コード: timm/models/xcit.py · アーキテクチャ: 28 · 事前学習済み重み: 42 (ImageNet-1kの結果付き42)

Cross-Covariance Image Transformers (XCiT) は、トークン間の自己注意を、特徴チャネル間のクロス共分散注意に置き換えます。この注意はトークン数に対して線形であり、ローカルパッチ相互作用レイヤーを追加します。

論文

使用方法

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

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

モデルAPIのサポート

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

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

アーキテクチャ

アーキテクチャパラメータ数 (M)GMACs入力サイズ事前学習済み重み
xcit_large_24_p8_224188.93141.232242
xcit_large_24_p8_384188.93415.003841
xcit_large_24_p16_224189.1035.862242
xcit_large_24_p16_384189.10105.353841
xcit_medium_24_p8_22484.3263.532242
xcit_medium_24_p8_38484.32186.673841
xcit_medium_24_p16_22484.4016.132242
xcit_medium_24_p16_38484.4047.393841
xcit_nano_12_p8_2243.052.162242
xcit_nano_12_p8_3843.056.343841
xcit_nano_12_p16_2243.050.562242
xcit_nano_12_p16_3843.051.643841
xcit_small_12_p8_22426.2118.692242
xcit_small_12_p8_38426.2154.923841
xcit_small_12_p16_22426.254.822242
xcit_small_12_p16_38426.2514.143841
xcit_small_24_p8_22447.6335.812242
xcit_small_24_p8_38447.63105.243841
xcit_small_24_p16_22447.679.102242
xcit_small_24_p16_38447.6726.723841
xcit_tiny_12_p8_2246.714.812242
xcit_tiny_12_p8_3846.7114.133841
xcit_tiny_12_p16_2246.721.242242
xcit_tiny_12_p16_3846.723.643841
xcit_tiny_24_p8_22412.119.212242
xcit_tiny_24_p8_38412.1127.053841
xcit_tiny_24_p16_22412.122.342242
xcit_tiny_24_p16_38412.126.873841

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

事前学習済み重み

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

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
xcit_large_24_p8_224.fb_dist_in1k22485.3962241000apache-2.0timm
xcit_large_24_p8_224.fb_in1k22484.3802241000apache-2.0timm
xcit_large_24_p8_384.fb_dist_in1k38486.0243841000apache-2.0timm
xcit_large_24_p16_224.fb_dist_in1k22484.9362241000apache-2.0timm
xcit_large_24_p16_224.fb_in1k22482.8862241000apache-2.0timm
xcit_large_24_p16_384.fb_dist_in1k38485.7783841000apache-2.0timm
xcit_medium_24_p8_224.fb_dist_in1k22485.0782241000apache-2.0timm
xcit_medium_24_p8_224.fb_in1k22483.7442241000apache-2.0timm
xcit_medium_24_p8_384.fb_dist_in1k38485.8423841000apache-2.0timm
xcit_medium_24_p16_224.fb_dist_in1k22484.2922241000apache-2.0timm
xcit_medium_24_p16_224.fb_in1k22482.6302241000apache-2.0timm
xcit_medium_24_p16_384.fb_dist_in1k38485.4343841000apache-2.0timm
xcit_nano_12_p8_224.fb_dist_in1k22476.3302241000apache-2.0timm
xcit_nano_12_p8_224.fb_in1k22473.9002241000apache-2.0timm
xcit_nano_12_p8_384.fb_dist_in1k38477.8103841000apache-2.0timm
xcit_nano_12_p16_224.fb_dist_in1k22472.3182241000apache-2.0timm
xcit_nano_12_p16_224.fb_in1k22469.9822241000apache-2.0timm
xcit_nano_12_p16_384.fb_dist_in1k38475.4603841000apache-2.0timm
xcit_small_12_p8_224.fb_dist_in1k22484.2522241000apache-2.0timm
xcit_small_12_p8_224.fb_in1k22483.3542241000apache-2.0timm
xcit_small_12_p8_384.fb_dist_in1k38485.0683841000apache-2.0timm
xcit_small_12_p16_224.fb_dist_in1k22483.3542241000apache-2.0timm
xcit_small_12_p16_224.fb_in1k22481.9822241000apache-2.0timm
xcit_small_12_p16_384.fb_dist_in1k38484.7343841000apache-2.0timm
xcit_small_24_p8_224.fb_dist_in1k22484.8782241000apache-2.0timm
xcit_small_24_p8_224.fb_in1k22483.8542241000apache-2.0timm
xcit_small_24_p8_384.fb_dist_in1k38485.5803841000apache-2.0timm
xcit_small_24_p16_224.fb_dist_in1k22483.8802241000apache-2.0timm
xcit_small_24_p16_224.fb_in1k22482.5882241000apache-2.0timm
xcit_small_24_p16_384.fb_dist_in1k38485.1023841000apache-2.0timm
xcit_tiny_12_p8_224.fb_dist_in1k22481.2142241000apache-2.0timm
xcit_tiny_12_p8_224.fb_in1k22479.7102241000apache-2.0timm
xcit_tiny_12_p8_384.fb_dist_in1k38482.3943841000apache-2.0timm
xcit_tiny_12_p16_224.fb_dist_in1k22478.5722241000apache-2.0timm
xcit_tiny_12_p16_224.fb_in1k22477.1222241000apache-2.0timm
xcit_tiny_12_p16_384.fb_dist_in1k38480.9543841000apache-2.0timm
xcit_tiny_24_p8_224.fb_dist_in1k22482.5782241000apache-2.0timm
xcit_tiny_24_p8_224.fb_in1k22481.8882241000apache-2.0timm
xcit_tiny_24_p8_384.fb_dist_in1k38483.7843841000apache-2.0timm
xcit_tiny_24_p16_224.fb_dist_in1k22480.4622241000apache-2.0timm
xcit_tiny_24_p16_224.fb_in1k22479.4742241000apache-2.0timm
xcit_tiny_24_p16_384.fb_dist_in1k38482.5803841000apache-2.0timm
Update on GitHub