timm documentation

CoaT

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

CoaT

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

Les Transformers d'images à attention convolutive multi-échelles (CoaT) utilisent une attention factorisée avec des encodages de position relative basés sur des convolutions, ainsi qu'un mécanisme multi-échelles qui échange des informations entre des blocs parallèles à différentes échelles.

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

Prise en charge de l'API des modèles

FonctionnalitéCoaT
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 (CoaT avec coat_tiny à 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
coat_lite_medium44.579.812241
coat_lite_medium_38444.5728.733841
coat_lite_mini11.012.002241
coat_lite_small19.843.962241
coat_lite_tiny5.721.602241
coat_mini10.346.822241
coat_small21.6912.612241
coat_tiny5.504.352241

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
coat_lite_medium.in1k22483.6022241000apache-2.0timm
coat_lite_medium_384.in1k38484.8903841000apache-2.0timm
coat_lite_mini.in1k22479.1042241000apache-2.0timm
coat_lite_small.in1k22482.3042241000apache-2.0timm
coat_lite_tiny.in1k22477.5482241000apache-2.0timm
coat_mini.in1k22481.3202241000apache-2.0timm
coat_small.in1k22482.3782241000apache-2.0timm
coat_tiny.in1k22478.4422241000apache-2.0timm
Update on GitHub