timm documentation

NaFlex ViT

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

and get access to the augmented documentation experience

to get started

NaFlex ViT

Categoría: ViT · Código: timm/models/naflexvit.py · Arquitecturas: 14 · Pesos preentrenados: 11 (3 con resultados en ImageNet-1k)

Un Vision Transformer con el incrustado de parches y la codificación de posición en un solo módulo, que admite entradas pre-parcheadas con relación de aspecto y resolución variables (NaFlex), tamaño de parche variable de FlexiViT y incrustamientos de posición factorizados de NaViT.

Papers

Referenciado por pesos preentrenados:

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

Soporte de la API de modelos

CaracterísticaNaFlexVit
mapas de características de features_only=True✓ (pasos 16, 16, 16)
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 (NaFlexVit con naflexvit_base_patch16_parfac_gap en 384); 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
naflexvit_base_patch16_gap86.6355.863841
naflexvit_base_patch16_map93.7256.233840
naflexvit_base_patch16_par_gap86.6355.863841
naflexvit_base_patch16_parfac_gap86.4655.863841
naflexvit_base_patch16_radio_v398.25-5121
naflexvit_base_patch16_siglip92.9356.123841
naflexvit_giant_patch16_radio_v31159.62-5121
naflexvit_huge_patch16_radio_v3651.64-5121
naflexvit_huge_patch16_radio_v4651.65-5121
naflexvit_large_patch16_radio_v3319.93-5121
naflexvit_so150m2_patch16_reg1_gap136.0689.533840
naflexvit_so150m2_patch16_reg1_map142.4690.333840
naflexvit_so400m_patch16_radio_v4431.24-5121
naflexvit_so400m_patch16_siglip427.89259.653841

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.

arXiv:2412.07679

Las configuraciones preentrenadas hacen referencia a arXiv:2412.07679.

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
naflexvit_base_patch16_radio_v3.c_radio512--nonenvidia-open-model-licensetimm
naflexvit_giant_patch16_radio_v3.c_radio512--nonenvidia-open-model-licensetimm
naflexvit_huge_patch16_radio_v3.c_radio512--nonenvidia-open-model-licensetimm
naflexvit_large_patch16_radio_v3.c_radio512--nonenvidia-open-model-licensetimm

etiquetas v2

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
naflexvit_base_patch16_siglip.v2_webli384--noneapache-2.0timm
naflexvit_so400m_patch16_siglip.v2_webli384--noneapache-2.0timm

arXiv:2601.17237

Las configuraciones preentrenadas hacen referencia a arXiv:2601.17237.

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
naflexvit_huge_patch16_radio_v4.c_radio512--nonenvidia-open-model-licensetimm
naflexvit_so400m_patch16_radio_v4.c_radio512--nonenvidia-open-model-licensetimm

Otros

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
naflexvit_base_patch16_gap.e300_s576_in1k38483.3403841000apache-2.0timm
naflexvit_base_patch16_par_gap.e300_s576_in1k38483.5083841000apache-2.0timm
naflexvit_base_patch16_parfac_gap.e300_s576_in1k38483.4683841000apache-2.0timm
Update on GitHub