timm documentation
PNASNet
はじめに
チュートリアル
timmのモデルAPIモデルの適応とエクスポート事前学習モデルを特徴抽出器として使用ハイパーパラメータ (HParams)公式トレーニングスクリプトの使用🤗 Hugging Face Hub からモデルを共有して読み込む
モデルページ
モデルファミリー結果
畳み込みネットワーク
Aligned XceptionBYOBNet (GENet、RepVGG、MobileOne など)ConvMixerConvNeXtCSPNet(CSPResNet、CSPDarkNet、DarkNet)CSPNeXtDenseNetDLADPNHRNetInception-ResNet-v2Inception-v3Inception-v4InceptionNeXtMambaOutNASNetNFNetPNASNetPP-HGNetRDNetRegNetRes2NetResNeStResNetResNetV2 (BiT)SelecSLSSENet (レガシー)SKNetTResNetVGGVoVNetXception (レガシー)
Vision Transformers
ハイブリッドな畳み込み / 注意機構ネットワーク
効率的なモバイルネットワーク
MLPおよびその他のトークンミキサー
マルチモーダルモデルのビジョンエンコーダー
リファレンス
PNASNet
カテゴリ: CNN · コード: timm/models/pnasnet.py · アーキテクチャ: 1 · 事前学習済み重み: 1 (ImageNet-1kの結果付き1)
PNASNet-5 Largeは、漸進的ニューラルアーキテクチャサーチによって発見されたモデルで、Cadeneの事前学習済みモデル移植版から提供されています。
論文
使用方法
事前学習済み重みの読み込み、画像の前処理、ファインチューニングについてはクイックスタートを、モデルをバックボーンとして使用する方法については特徴量抽出を参照してください。
>>> import timm
>>> model = timm.create_model('pnasnet5large.tf_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)モデルAPIのサポート
| 機能 | PNASNet5Large |
|---|---|
features_only=True 特徴マップ | ✓ (strides 2, 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) 正方形サイズのみ動作、✗ 非対応、– 不要(モデルはすでに任意の入力サイズを受け入れる)。ランダム初期化されたモデル(pnasnet5large の PNASNet5Large、331)で各機能を実行して確認しています。同じクラスの他のアーキテクチャも同様に動作すると予想されます。詳細はModel APIページを参照してください。
アーキテクチャ
| アーキテクチャ | パラメータ数 (M) | GMACs | 入力サイズ | 事前学習済み重み |
|---|---|---|---|---|
pnasnet5large | 86.06 | 25.04 | 331 | 1 |
GMACsとパラメータ数は、各アーキテクチャのデフォルトの入力サイズと分類器に基づいており、ベンチマーク結果から得られたものです。
事前学習済み重み
Top-1 は、results-imagenet.csv に記載された評価サイズでの ImageNet-1k 検証精度です。- は、ImageNet-1k の結果がない重み(例: プリトレーニング用または画像-テキスト用重み)を示します。Classes は事前学習済み分類器のサイズで、分類器を持たない重みについては none です。
| 重み | 入力サイズ | Top-1 (%) | 評価サイズ | クラス | ライセンス | Hub |
|---|---|---|---|---|---|---|
pnasnet5large.tf_in1k | 331 | 82.814 | 331 | 1000 | apache-2.0 | timm |