timm documentation

SKNet

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

and get access to the augmented documentation experience

to get started

SKNet

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

Redes Selective Kernel sobre una base ResNet, donde las unidades SK fusionan ramas con diferentes tamaños de kernel mediante atención sobre las ramas.

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

Soporte de la API de modelos

CaracterísticaResNet
mapas de características de features_only=True✓ (pasos 2, 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 (ResNet con skresnet18 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
skresnet1811.961.822241
skresnet3422.283.672241
skresnet5025.804.112240
skresnet50d25.824.362240
skresnext50_32x4d27.484.502241

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
skresnet18.ra_in1k22473.0202241000apache-2.0timm
skresnet34.ra_in1k22476.9562241000apache-2.0timm
skresnext50_32x4d.ra_in1k22480.1682241000apache-2.0timm
Update on GitHub