timm documentation

PP-HGNet

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.31).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

PP-HGNet

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

PP-HGNet et PP-HGNetV2 (High Performance GPU Net) de PaddleClas, construits à partir de blocs HG qui agrègent les sorties d'une séquence de convolutions. Les poids ssld sont entraînés avec la distillation SSLD de PaddleClas.

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('hgnet_small.paddle_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Prise en charge de l'API des modèles

FonctionnalitéHighPerfGpuNet
cartes de caractéristiques features_only=True✓ (pas 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 (HighPerfGpuNet avec hgnetv2_b0 à 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
hgnet_base71.5825.142241
hgnet_small24.368.532242
hgnet_tiny14.744.542242
hgnetv2_b06.000.332242
hgnetv2_b16.340.492242
hgnetv2_b211.221.152242
hgnetv2_b316.291.782242
hgnetv2_b419.802.752242
hgnetv2_b539.576.562242
hgnetv2_b675.2616.882242

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.

tags ssld

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
hgnet_base.ssld_in1k224 (test 288)85.4882881000apache-2.0timm
hgnet_small.ssld_in1k224 (test 288)84.3682881000apache-2.0timm
hgnet_tiny.ssld_in1k224 (test 288)82.5282881000apache-2.0timm
hgnetv2_b0.ssld_stage1_in22k_in1k224 (test 288)78.0382881000apache-2.0timm
hgnetv2_b0.ssld_stage2_ft_in1k224 (test 288)78.5902881000apache-2.0timm
hgnetv2_b1.ssld_stage1_in22k_in1k224 (test 288)79.0502881000apache-2.0timm
hgnetv2_b1.ssld_stage2_ft_in1k224 (test 288)79.9162881000apache-2.0timm
hgnetv2_b2.ssld_stage1_in22k_in1k224 (test 288)81.4562881000apache-2.0timm
hgnetv2_b2.ssld_stage2_ft_in1k224 (test 288)82.3622881000apache-2.0timm
hgnetv2_b3.ssld_stage1_in22k_in1k224 (test 288)83.1182881000apache-2.0timm
hgnetv2_b3.ssld_stage2_ft_in1k224 (test 288)83.5882881000apache-2.0timm
hgnetv2_b4.ssld_stage1_in22k_in1k224 (test 288)83.4502881000apache-2.0timm
hgnetv2_b4.ssld_stage2_ft_in1k224 (test 288)83.9522881000apache-2.0timm
hgnetv2_b5.ssld_stage1_in22k_in1k224 (test 288)84.9302881000apache-2.0timm
hgnetv2_b5.ssld_stage2_ft_in1k224 (test 288)85.1582881000apache-2.0timm
hgnetv2_b6.ssld_stage1_in22k_in1k224 (test 288)86.3002881000apache-2.0timm
hgnetv2_b6.ssld_stage2_ft_in1k224 (test 288)86.3802881000apache-2.0timm

tags paddle

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
hgnet_small.paddle_in1k224 (test 288)82.2222881000apache-2.0timm
hgnet_tiny.paddle_in1k224 (test 288)80.6442881000apache-2.0timm
Update on GitHub