timm documentation

RegNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

RegNet

カテゴリ: CNN · コード: timm/models/regnet.py · アーキテクチャ: 36 · 事前学習済み重み: 65 (うち60がImageNet-1kの結果あり)

regular network design spaces 由来の RegNetX と RegNetY ネットワーク、“Fast and Accurate Model Scaling” 由来の RegNetZ、および pre-activation RegNetV などの timm による追加実装を含みます。pycls、torchvision、VISSL (SEER) からの重みも含まれます。

論文

使用方法

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

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

モデルAPIのサポート

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

アーキテクチャ

アーキテクチャパラメータ数 (M)GMACs入力サイズ事前学習済み重み
regnetv_04020.644.002241
regnetv_06430.586.392241
regnetx_0022.680.202241
regnetx_0045.160.402241
regnetx_004_tv5.500.422241
regnetx_0066.200.612241
regnetx_0087.260.812242
regnetx_0169.191.622242
regnetx_03215.303.202242
regnetx_04022.123.992241
regnetx_06426.216.492241
regnetx_08039.578.022242
regnetx_12046.1112.132241
regnetx_16054.2815.992242
regnetx_320107.8131.812242
regnety_0023.160.202241
regnety_0044.340.412242
regnety_0066.060.612241
regnety_0086.260.812241
regnety_008_tv6.430.842241
regnety_01611.201.632242
regnety_03219.443.202243
regnety_04020.654.002242
regnety_040_sgn20.654.032240
regnety_06430.586.392242
regnety_08039.188.002242
regnety_080_tv39.388.512241
regnety_12051.8212.142243
regnety_16083.5915.962248
regnety_320145.0532.342246
regnety_640281.38188.473842
regnety_1280644.81374.993844
regnety_25601282.60747.833841
regnetz_0057.120.522240
regnetz_04027.124.062561
regnetz_040_h28.944.122561

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

事前学習済み重み

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

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

facebookresearch/pycls

出典: github.com/facebookresearch/pycls

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
regnetx_002.pycls_in1k22468.7582241000mittimm
regnetx_004.pycls_in1k22472.4822241000mittimm
regnetx_006.pycls_in1k22473.9042241000mittimm
regnetx_008.pycls_in1k22475.0542241000mittimm
regnetx_016.pycls_in1k22476.9362241000mittimm
regnetx_032.pycls_in1k22478.1602241000mittimm
regnetx_040.pycls_in1k22478.4862241000mittimm
regnetx_064.pycls_in1k22479.0622241000mittimm
regnetx_080.pycls_in1k22479.2142241000mittimm
regnetx_120.pycls_in1k22479.6082241000mittimm
regnetx_160.pycls_in1k22479.8442241000mittimm
regnetx_320.pycls_in1k22480.2582241000mittimm
regnety_002.pycls_in1k22470.2882241000mittimm
regnety_004.pycls_in1k22474.0042241000mittimm
regnety_006.pycls_in1k22475.2742241000mittimm
regnety_008.pycls_in1k22476.3342241000mittimm
regnety_016.pycls_in1k22477.8802241000mittimm
regnety_032.pycls_in1k22478.9022241000mittimm
regnety_040.pycls_in1k22479.2462241000mittimm
regnety_064.pycls_in1k22479.7342241000mittimm
regnety_080.pycls_in1k22479.8742241000mittimm
regnety_120.pycls_in1k22480.3822241000mittimm
regnety_160.pycls_in1k22480.2842241000mittimm
regnety_320.pycls_in1k22480.7962241000mittimm

pytorch/vision

出典: github.com/pytorch/vision

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
regnetx_004_tv.tv2_in1k22474.5962241000bsd-3-clausetimm
regnetx_008.tv2_in1k22477.2942241000bsd-3-clausetimm
regnetx_016.tv2_in1k22479.4402241000bsd-3-clausetimm
regnetx_032.tv2_in1k22480.9162241000bsd-3-clausetimm
regnetx_080.tv2_in1k22481.5522241000bsd-3-clausetimm
regnetx_160.tv2_in1k22482.5742241000bsd-3-clausetimm
regnetx_320.tv2_in1k22482.8242241000bsd-3-clausetimm
regnety_004.tv2_in1k22475.5942241000bsd-3-clausetimm
regnety_008_tv.tv2_in1k22478.6722241000bsd-3-clausetimm
regnety_016.tv2_in1k22480.6682241000bsd-3-clausetimm
regnety_032.tv2_in1k22481.7882241000bsd-3-clausetimm
regnety_080_tv.tv2_in1k22482.5802241000bsd-3-clausetimm
regnety_160.swag_ft_in1k38486.0203841000cc-by-nc-4.0timm
regnety_160.swag_lc_in1k22483.8062241000cc-by-nc-4.0timm
regnety_160.tv2_in1k22482.6462241000bsd-3-clausetimm
regnety_320.swag_ft_in1k38486.8303841000cc-by-nc-4.0timm
regnety_320.swag_lc_in1k22484.6542241000cc-by-nc-4.0timm
regnety_320.tv2_in1k22483.1702241000bsd-3-clausetimm
regnety_1280.swag_ft_in1k38488.2283841000cc-by-nc-4.0timm
regnety_1280.swag_lc_in1k22485.9642241000cc-by-nc-4.0timm

ra3 タグ

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
regnetv_040.ra3_in1k224(テスト 288)83.2042881000apache-2.0timm
regnetv_064.ra3_in1k224(テスト 288)83.7462881000apache-2.0timm
regnety_040.ra3_in1k224(テスト 288)83.0482881000apache-2.0timm
regnety_064.ra3_in1k224(テスト 288)83.7162881000apache-2.0timm
regnety_080.ra3_in1k224(テスト 288)83.9402881000apache-2.0timm
regnetz_040.ra3_in1k256(テスト 320)84.2423201000apache-2.0timm
regnetz_040_h.ra3_in1k256(テスト 320)84.4983201000apache-2.0timm

facebookresearch/vissl

出典: github.com/facebookresearch/vissl

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
regnety_320.seer224--なしseer-licensetimm
regnety_320.seer_ft_in1k38483.3243841000seer-licensetimm
regnety_640.seer224--なしseer-licensetimm
regnety_640.seer_ft_in1k38483.9243841000seer-licensetimm
regnety_1280.seer224--なしseer-licensetimm
regnety_1280.seer_ft_in1k38484.4263841000seer-licensetimm
regnety_2560.seer_ft_in1k38485.1443841000seer-licensetimm

sw タグ

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
regnety_120.sw_in12k224(テスト 288)--11821apache-2.0timm
regnety_120.sw_in12k_ft_in1k224(テスト 288)85.4282881000apache-2.0timm
regnety_160.sw_in12k224(テスト 288)--11821apache-2.0timm
regnety_160.sw_in12k_ft_in1k224(テスト 288)86.0282881000apache-2.0timm

その他

重み入力サイズTop-1 (%)評価サイズクラスライセンスHub
regnety_032.ra_in1k224(テスト 288)82.7242881000apache-2.0timm
regnety_160.deit_in1k224(テスト 288)83.7042881000apache-2.0timm
regnety_160.lion_in12k_ft_in1k224(テスト 288)86.0042881000apache-2.0timm
Update on GitHub