timm documentation

ConvMixer

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

ConvMixer

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

ConvMixer incrusta la imagen como parches y mantiene esa resolución durante todo el proceso, alternando entre convoluciones depthwise (mezcla espacial) y convoluciones pointwise (mezcla de canales).

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

Soporte de la API de modelos

CaracterísticaConvMixer
mapas de características de features_only=True✓ (pasos 7, 7, 7)
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 (ConvMixer con convmixer_768_32 a 224), se espera que otras arquitecturas de la misma clase se comporten de la misma manera. Consulte la página Model API para obtener más detalles.

Arquitecturas

ArquitecturaParámetros (M)GMACsTamaño de entradaPesos preentrenados
convmixer_768_3221.1119.552241
convmixer_1024_20_ks9_p1424.385.552241
convmixer_1536_2051.6348.682241

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
convmixer_768_32.in1k22480.1582241000mittimm
convmixer_1024_20_ks9_p14.in1k22476.9442241000mittimm
convmixer_1536_20.in1k22481.3822241000mittimm
Update on GitHub