timm documentation

Hiera

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Hiera

Categoría: ViT · Código: timm/models/hiera.py · Arquitecturas: 8 · Pesos preentrenados: 16 (8 con resultados en ImageNet-1k)

Hiera es un transformer jerárquico de visión sin módulos especializados (convoluciones, ventanas desplazadas, sesgos de posición relativos), que se apoya en el preentrenamiento con autoencoder enmascarado (MAE) para aprender sesgos espaciales.

Papers

Uso

Consulte la Introducción rápida para cargar pesos preentrenados, preprocesar imágenes y realizar el ajuste fino, y Extracción de características para usar los modelos como columnas verticales.

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

Soporte de la API de modelos

CaracterísticaHiera
mapas de características de features_only=True✓ (pasos 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Cualquier tamaño de entrada, no se necesitan cambios✗
set_input_size()✗
dynamic_img_size=True✗
set_grad_checkpointing()✓
Decaimiento del LR por capa (group_matcher())✓
reset_classifier()✓

✓ compatible, ✓ (cuadrado) funciona solo con tamaños cuadrados, ✗ no compatible, – no necesario (el modelo ya acepta cualquier tamaño de entrada). Verificado ejecutando cada función en un modelo inicializado aleatoriamente (Hiera con hiera_tiny_224 en 224); se espera que otras arquitecturas de la misma clase se comporten de la misma manera. Consulte la página Model API para más detalles.

Arquitecturas

ArquitecturaParámetros (M)GMACsTamaño de entradaPesos preentrenados
hiera_base_22451.529.402242
hiera_base_abswin_25651.2712.462560
hiera_base_plus_22469.9012.672242
hiera_huge_224672.78124.852242
hiera_large_224213.7440.342242
hiera_small_22435.016.422242
hiera_small_abswin_25634.368.292564
hiera_tiny_22427.914.912242

Los GMAC y el número de parámetros corresponden al tamaño de entrada y al clasificador predeterminados de cada arquitectura, según los resultados de las pruebas de rendimiento.

Pesos preentrenados

Top-1 es la precisión de validación en ImageNet-1k desde results-imagenet.csv en el tamaño de evaluación indicado, - si el peso no tiene un resultado de ImageNet-1k (p. ej. pesos de preentrenamiento o de imagen-texto). Classes es el tamaño del clasificador preentrenado, ninguno para pesos sin uno.

Los pesos se agrupan por origen: el artículo o procedencia indicado en la configuración preentrenada, en su defecto una organización del Hub distinta de timm, y si no, la primera parte de la etiqueta preentrenada (architecture.tag). Los orígenes con un solo peso se listan bajo other.

etiquetas de mae

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
hiera_base_224.mae224--nonecc-by-nc-4.0timm
hiera_base_224.mae_in1k_ft_in1k22484.5282241000cc-by-nc-4.0timm
hiera_base_plus_224.mae224--nonecc-by-nc-4.0timm
hiera_base_plus_224.mae_in1k_ft_in1k22485.1362241000cc-by-nc-4.0timm
hiera_huge_224.mae224--nonecc-by-nc-4.0timm
hiera_huge_224.mae_in1k_ft_in1k22486.8402241000cc-by-nc-4.0timm
hiera_large_224.mae224--nonecc-by-nc-4.0timm
hiera_large_224.mae_in1k_ft_in1k22486.0462241000cc-by-nc-4.0timm
hiera_small_224.mae224--nonecc-by-nc-4.0timm
hiera_small_224.mae_in1k_ft_in1k22483.8882241000cc-by-nc-4.0timm
hiera_tiny_224.mae224--nonecc-by-nc-4.0timm
hiera_tiny_224.mae_in1k_ft_in1k22482.8262241000cc-by-nc-4.0timm

etiquetas sbb2

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
hiera_small_abswin_256.sbb2_e200_in12k256--11821apache-2.0timm
hiera_small_abswin_256.sbb2_e200_in12k_ft_in1k25684.9122561000apache-2.0timm
hiera_small_abswin_256.sbb2_pd_e200_in12k256--11821apache-2.0timm
hiera_small_abswin_256.sbb2_pd_e200_in12k_ft_in1k25684.5642561000apache-2.0timm
Update on GitHub