timm documentation

RepGhost

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

RepGhost

Catégorie : Efficient/Mobile · Code : timm/models/repghost.py · Architectures : 8 · Poids pré-entraînés : 8 (8 avec des résultats ImageNet-1k)

RepGhostNet remplace la concaténation de caractéristiques du module Ghost par une réutilisation de caractéristiques reparamétrée, intégrée dans un seul chemin de convolution lors de l'inférence.

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

Prise en charge de l'API des modèles

FonctionnalitéRepGhostNet
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 (RepGhostNet avec repghostnet_050 à 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
repghostnet_0502.310.052241
repghostnet_0582.540.062241
repghostnet_0803.270.102241
repghostnet_1004.060.152241
repghostnet_1114.520.182241
repghostnet_1305.460.242241
repghostnet_1506.550.312241
repghostnet_2009.770.532241

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.

PoidsTaille d'entréeTop-1 (%)Taille d'évaluationClassesLicenceHub
repghostnet_050.in1k22466.9782241000mittimm
repghostnet_058.in1k22468.9382241000mittimm
repghostnet_080.in1k22472.2402241000mittimm
repghostnet_100.in1k22474.2082241000mittimm
repghostnet_111.in1k22475.0602241000mittimm
repghostnet_130.in1k22476.3782241000mittimm
repghostnet_150.in1k22477.4742241000mittimm
repghostnet_200.in1k22478.8022241000mittimm
Update on GitHub