timm documentation
Swin Transformer V2
Swin Transformer V2
Categoría: ViT · Código: timm/models/swin_transformer_v2.py · Arquitecturas: 12 · Pesos preentrenados: 12 (10 con resultados en ImageNet-1k)
Swin Transformer V2 escala Swin con normalización residual posterior, atención coseno escalada y un sesgo de posición relativa continua espaciado en escala logarítmica que se transfiere entre tamaños de ventana.
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_tiny_window16_256.ms_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)Soporte de la API de modelos
✓ 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 (SwinTransformerV2 con swinv2_tiny_window16_256 en 256); 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
| Arquitectura | Parámetros (M) | GMACs | Tamaño de entrada | Pesos preentrenados |
|---|---|---|---|---|
swinv2_base_window8_256 | 87.92 | 20.37 | 256 | 1 |
swinv2_base_window12_192 | 109.28 | 11.90 | 192 | 1 |
swinv2_base_window12to16_192to256 | 87.92 | 22.02 | 256 | 1 |
swinv2_base_window12to24_192to384 | 87.92 | 55.25 | 384 | 1 |
swinv2_base_window16_256 | 87.92 | 22.02 | 256 | 1 |
swinv2_large_window12_192 | 228.77 | 26.17 | 192 | 1 |
swinv2_large_window12to16_192to256 | 196.74 | 47.81 | 256 | 1 |
swinv2_large_window12to24_192to384 | 196.74 | 116.15 | 384 | 1 |
swinv2_small_window8_256 | 49.73 | 11.58 | 256 | 1 |
swinv2_small_window16_256 | 49.73 | 12.82 | 256 | 1 |
swinv2_tiny_window8_256 | 28.35 | 5.96 | 256 | 1 |
swinv2_tiny_window16_256 | 28.35 | 6.68 | 256 | 1 |
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.
| Peso | Tamaño de entrada | Top-1 (%) | Tamaño de evaluación | Clases | Licencia | Hub |
|---|---|---|---|---|---|---|
swinv2_base_window8_256.ms_in1k | 256 | 84.246 | 256 | 1000 | mit | timm |
swinv2_base_window12_192.ms_in22k | 192 | - | - | 21841 | mit | timm |
swinv2_base_window12to16_192to256.ms_in22k_ft_in1k | 256 | 86.270 | 256 | 1000 | mit | timm |
swinv2_base_window12to24_192to384.ms_in22k_ft_in1k | 384 | 87.142 | 384 | 1000 | mit | timm |
swinv2_base_window16_256.ms_in1k | 256 | 84.608 | 256 | 1000 | mit | timm |
swinv2_large_window12_192.ms_in22k | 192 | - | - | 21841 | mit | timm |
swinv2_large_window12to16_192to256.ms_in22k_ft_in1k | 256 | 86.934 | 256 | 1000 | mit | timm |
swinv2_large_window12to24_192to384.ms_in22k_ft_in1k | 384 | 87.474 | 384 | 1000 | mit | timm |
swinv2_small_window8_256.ms_in1k | 256 | 83.844 | 256 | 1000 | mit | timm |
swinv2_small_window16_256.ms_in1k | 256 | 84.216 | 256 | 1000 | mit | timm |
swinv2_tiny_window8_256.ms_in1k | 256 | 81.824 | 256 | 1000 | mit | timm |
swinv2_tiny_window16_256.ms_in1k | 256 | 82.828 | 256 | 1000 | mit | timm |