timm documentation

VoVNet

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

VoVNet

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

VoVNet utilise des modules One-Shot Aggregation (OSA), qui concatènent les sorties de toutes les couches d’un bloc une seule fois à la fin de celui-ci. VoVNetV2 (issu de CenterMask) ajoute des connexions résiduelles et une attention SE efficace (eSE).

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

Prise en charge de l'API des modèles

FonctionnalitéVovNet
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 (VovNet avec ese_vovnet19b_slim_dw à 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
eca_vovnet39b22.607.092240
ese_vovnet19b_dw6.541.342241
ese_vovnet19b_slim3.171.692240
ese_vovnet19b_slim_dw1.900.402240
ese_vovnet39b24.577.092241
ese_vovnet39b_evos24.587.072240
ese_vovnet57b38.6111.692561
ese_vovnet99b63.2016.512240
vovnet39a22.607.092240
vovnet57a36.648.952240

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
ese_vovnet19b_dw.ra_in1k224 (test 288)77.7882881000apache-2.0timm
ese_vovnet39b.ra_in1k224 (test 288)80.3782881000apache-2.0timm
ese_vovnet57b.ra4_e3600_r256_in1k256 (test 320)83.2983201000apache-2.0timm
Update on GitHub