timm documentation

NASNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

NASNet

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

Cadeneの事前学習済みモデル移植版である、ニューラルアーキテクチャサーチによって発見されたセルのネットワーク、NASNet-A Large。

論文

使用方法

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

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

モデルAPIのサポート

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

アーキテクチャ

アーキテクチャパラメータ数 (M)GMACs入力サイズ事前学習済み重み
nasnetalarge88.7523.893311

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

事前学習済み重み

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

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
nasnetalarge.tf_in1k33182.6383311000apache-2.0timm
Update on GitHub