timm documentation

RegNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

RegNet

Catégorie : CNN · Code : timm/models/regnet.py · Architectures : 36 · Poids pré-entraînés : 65 (60 avec des résultats ImageNet-1k)

Les réseaux RegNetX et RegNetY des espaces de conception de réseaux réguliers, RegNetZ issu de « Fast and Accurate Model Scaling », et les ajouts de timm tels que RegNetV pré-activation. Inclut les poids de pycls, torchvision et VISSL (SEER).

Articles

Utilisation

Consultez le Démarrage rapide pour le chargement des poids pré-entraînés, le prétraitement des images et l’ajustement fin, ainsi que Extraction de caractéristiques pour utiliser les modèles comme squelettes.

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

Prise en charge de l'API des modèles

FonctionnalitéRegNet
cartes de caractéristiques features_only=True✓ (pas 2, 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Toute taille d'entrée, aucune modification nécessaire✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Décroissance du taux d'apprentissage par couche (group_matcher())✓
reset_classifier()✓

✓ pris en charge, ✓ (carré) fonctionne uniquement pour les tailles carrées, ✗ non pris en charge, – non nécessaire (le modèle accepte déjà n'importe quelle taille d'entrée). Vérifié en exécutant chaque fonctionnalité sur un modèle initialisé aléatoirement (RegNet avec regnetx_002 à 224), les autres architectures de la même classe sont censées se comporter de la même manière. Consultez la page API du modèle pour plus de détails.

Architectures

ArchitectureParamètres (M)GMACsTaille d'entréePoids pré-entraînés
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

Les GMACs et le nombre de paramètres correspondent à la taille d’entrée par défaut et au classifieur de chaque architecture, d’après les résultats des benchmarks.

Poids pré-entraînés

Top-1 est la précision de validation sur ImageNet-1k issue de results-imagenet.csv à la taille d'évaluation indiquée, - si le poids n'a pas de résultat ImageNet-1k (par exemple, poids de pré-entraînement ou poids image-texte). Classes est la taille du classifieur pré-entraîné, aucune pour les poids sans classifieur.

Les poids sont regroupés par source : le papier ou l’origine indiqué dans la configuration pré-entraînée, à défaut une organisation du Hub autre que timm, à défaut la première partie de l’étiquette pré-entraînée (architecture.tag). Les sources ne comportant qu’un seul poids sont listées sous other.

facebookresearch/pycls

Origine : github.com/facebookresearch/pycls

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
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

Origine : github.com/pytorch/vision

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
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

tags ra3

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
regnetv_040.ra3_in1k224 (test 288)83.2042881000apache-2.0timm
regnetv_064.ra3_in1k224 (test 288)83.7462881000apache-2.0timm
regnety_040.ra3_in1k224 (test 288)83.0482881000apache-2.0timm
regnety_064.ra3_in1k224 (test 288)83.7162881000apache-2.0timm
regnety_080.ra3_in1k224 (test 288)83.9402881000apache-2.0timm
regnetz_040.ra3_in1k256 (test 320)84.2423201000apache-2.0timm
regnetz_040_h.ra3_in1k256 (test 320)84.4983201000apache-2.0timm

facebookresearch/vissl

Origine : github.com/facebookresearch/vissl

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
regnety_320.seer224--noneseer-licensetimm
regnety_320.seer_ft_in1k38483.3243841000seer-licensetimm
regnety_640.seer224--noneseer-licensetimm
regnety_640.seer_ft_in1k38483.9243841000seer-licensetimm
regnety_1280.seer224--noneseer-licensetimm
regnety_1280.seer_ft_in1k38484.4263841000seer-licensetimm
regnety_2560.seer_ft_in1k38485.1443841000seer-licensetimm

sw tags

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
regnety_120.sw_in12k224 (test 288)--11821apache-2.0timm
regnety_120.sw_in12k_ft_in1k224 (test 288)85.4282881000apache-2.0timm
regnety_160.sw_in12k224 (test 288)--11821apache-2.0timm
regnety_160.sw_in12k_ft_in1k224 (test 288)86.0282881000apache-2.0timm

Autres

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
regnety_032.ra_in1k224 (test 288)82.7242881000apache-2.0timm
regnety_160.deit_in1k224 (test 288)83.7042881000apache-2.0timm
regnety_160.lion_in12k_ft_in1k224 (test 288)86.0042881000apache-2.0timm
Update on GitHub