timm documentation

Swin Transformer V2 (CR)

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Swin Transformer V2 (CR)

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

Una implementación alternativa de Swin Transformer V2 adaptada del código de Christoph Reich, diseñada para cambiar el tamaño de la imagen junto con el tamaño de la ventana, incluyendo tamaños no cuadrados. La docstring del módulo lo marca como experimental.

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

Soporte de la API de modelos

CaracterísticaSwinTransformerV2Cr
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 (SwinTransformerV2Cr con swinv2_cr_tiny_224 a 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
swinv2_cr_base_22487.8815.862240
swinv2_cr_base_38487.8850.573840
swinv2_cr_base_ns_22487.8815.862240
swinv2_cr_giant_2242598.76483.852240
swinv2_cr_giant_3842598.761450.713840
swinv2_cr_huge_224657.83115.972240
swinv2_cr_huge_384657.94352.043840
swinv2_cr_large_224196.6835.102240
swinv2_cr_large_384196.68108.963840
swinv2_cr_small_22449.709.072241
swinv2_cr_small_38449.7029.703840
swinv2_cr_small_ns_22449.709.082241
swinv2_cr_small_ns_25649.7012.072560
swinv2_cr_tiny_22428.334.662240
swinv2_cr_tiny_38428.3315.343840
swinv2_cr_tiny_ns_22428.334.662241

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.

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
swinv2_cr_small_224.sw_in1k22483.1262241000apache-2.0timm
swinv2_cr_small_ns_224.sw_in1k22483.4902241000apache-2.0timm
swinv2_cr_tiny_ns_224.sw_in1k22481.7962241000apache-2.0timm
Update on GitHub