timm documentation
Visformer
Visformer
Categoría: Híbrido · Código: timm/models/visformer.py · Arquitecturas: 2 · Pesos preentrenados: 2 (2 con resultados en ImageNet-1k)
Visformer (Vision-friendly Transformer) transiciona un transformer hacia un diseño convolucional, con bloques convolucionales en las primeras etapas y autoatención en las etapas posteriores.
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('visformer_small.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 (Visformer con visformer_tiny 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
| Arquitectura | Parámetros (M) | GMACs | Tamaño de entrada | Pesos preentrenados |
|---|---|---|---|---|
visformer_small | 40.22 | 4.88 | 224 | 1 |
visformer_tiny | 10.32 | 1.27 | 224 | 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 |
|---|---|---|---|---|---|---|
visformer_small.in1k | 224 | 82.092 | 224 | 1000 | apache-2.0 | timm |
visformer_tiny.in1k | 224 | 78.160 | 224 | 1000 | apache-2.0 | timm |