timm documentation
ツイン
はじめに
チュートリアル
timmのモデルAPIモデルの適応とエクスポート事前学習モデルを特徴抽出器として使用ハイパーパラメータ (HParams)公式トレーニングスクリプトの使用🤗 Hugging Face Hub からモデルを共有して読み込む
モデルページ
モデルファミリー結果
畳み込みネットワーク
Vision Transformers
BEiTCaiTConViTCrossViTDaViTDeiTEVAGCViTHieraHieraDet (SAM 2)MViTv2NaFlex ViTNesTPiTPVTv2Segment Anything ViTSwin TransformerSwin Transformer V2Swin Transformer V2 (CR)TNTツインVision Transformer(相対位置)Vision Transformer (ViT)VOLOXCiT
ハイブリッドな畳み込み / 注意機構ネットワーク
効率的なモバイルネットワーク
MLPおよびその他のトークンミキサー
マルチモーダルモデルのビジョンエンコーダー
リファレンス
ツイン
カテゴリ: ViT · コード: timm/models/twins.py · アーキテクチャ: 6 · 事前学習済み重み: 6 (ImageNet-1kの結果を含む6)
Twins-PCPVT(条件付き位置エンコーディング付き Pyramid Vision Transformer)と Twins-SVT(空間的に分離可能な自己注意、ローカルグループ化された注意とグローバルなサブサンプリングされた注意を交互に適用)。
論文
使用方法
事前学習済み重みの読み込み、画像の前処理、ファインチューニングについてはクイックスタートを、モデルをバックボーンとして使用する方法については特徴量抽出を参照してください。
>>> import timm
>>> model = timm.create_model('twins_pcpvt_small.in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)モデルAPIのサポート
✓ 対応、✓ (square) 正方形サイズのみ対応、✗ 非対応、– 不要(モデルはすでに任意の入力サイズを受け入れる)。各機能をランダムに初期化されたモデル(Twins の twins_svt_small、224)で実行して確認しています。同じクラスの他のアーキテクチャも同様に動作すると予想されます。詳細はModel APIページを参照してください。
アーキテクチャ
| アーキテクチャ | パラメータ数 (M) | GMACs | 入力サイズ | 事前学習済み重み |
|---|---|---|---|---|
twins_pcpvt_base | 43.83 | 6.68 | 224 | 1 |
twins_pcpvt_large | 60.99 | 9.84 | 224 | 1 |
twins_pcpvt_small | 24.11 | 3.83 | 224 | 1 |
twins_svt_base | 56.07 | 8.59 | 224 | 1 |
twins_svt_large | 99.27 | 15.15 | 224 | 1 |
twins_svt_small | 24.06 | 2.94 | 224 | 1 |
GMACsとパラメータ数は、各アーキテクチャのデフォルトの入力サイズと分類器に基づいており、ベンチマーク結果から得られたものです。
事前学習済み重み
Top-1 は、results-imagenet.csv に記載された評価サイズでの ImageNet-1k 検証精度です。- は、ImageNet-1k の結果がない重み(例: プリトレーニング用または画像-テキスト用重み)を示します。Classes は事前学習済み分類器のサイズで、分類器を持たない重みについては none です。
| 重み | 入力サイズ | Top-1 (%) | 評価サイズ | クラス | ライセンス | Hub |
|---|---|---|---|---|---|---|
twins_pcpvt_base.in1k | 224 | 82.724 | 224 | 1000 | apache-2.0 | timm |
twins_pcpvt_large.in1k | 224 | 83.158 | 224 | 1000 | apache-2.0 | timm |
twins_pcpvt_small.in1k | 224 | 81.116 | 224 | 1000 | apache-2.0 | timm |
twins_svt_base.in1k | 224 | 83.122 | 224 | 1000 | apache-2.0 | timm |
twins_svt_large.in1k | 224 | 83.718 | 224 | 1000 | apache-2.0 | timm |
twins_svt_small.in1k | 224 | 81.694 | 224 | 1000 | apache-2.0 | timm |