timm documentation

NFNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

NFNet

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

Redes Normalization-Free, entrenadas sin normalización por lotes mediante estandarización de pesos escalada: modelos NFNet (con recorte adaptativo de gradientes), NF-RegNet y NF-ResNet de pre-activación.

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

Soporte de la API de modelos

CaracterísticaNormFreeNet
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 (NormFreeNet con nf_regnet_b0 en 192); 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
dm_nfnet_f071.497.211921
dm_nfnet_f1132.6317.872241
dm_nfnet_f2193.7833.762561
dm_nfnet_f3254.9268.773201
dm_nfnet_f4316.07122.143841
dm_nfnet_f5377.21170.714161
dm_nfnet_f6438.36229.704481
eca_nfnet_l024.144.352241
eca_nfnet_l141.419.622561
eca_nfnet_l256.7220.953201
eca_nfnet_l372.0432.573520
nf_ecaresnet2616.002.412240
nf_ecaresnet5025.564.212240
nf_ecaresnet10144.558.012240
nf_regnet_b08.760.371920
nf_regnet_b110.220.822561
nf_regnet_b214.310.972400
nf_regnet_b318.591.672880
nf_regnet_b430.213.293200
nf_regnet_b549.747.953840
nf_resnet2616.002.412240
nf_resnet5025.565.462561
nf_resnet10144.558.012240
nf_seresnet2617.402.412240
nf_seresnet5028.094.212240
nf_seresnet10149.338.022240
nfnet_f071.497.211920
nfnet_f1132.6317.872240
nfnet_f2193.7833.762560
nfnet_f3254.9268.773200
nfnet_f4316.07122.143840
nfnet_f5377.21170.714160
nfnet_f6438.36229.704480
nfnet_f7499.50300.084800
nfnet_l035.074.362241
test_nfnet0.380.291601

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.

Los pesos se agrupan por origen: el artículo o procedencia indicado en la configuración preentrenada, en su defecto una organización del Hub distinta de timm, y si no, la primera parte de la etiqueta preentrenada (architecture.tag). Los orígenes con un solo peso se listan bajo other.

etiquetas de dm

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
dm_nfnet_f0.dm_in1k192 (prueba 256)83.4682561000apache-2.0timm
dm_nfnet_f1.dm_in1k224 (prueba 320)84.7003201000apache-2.0timm
dm_nfnet_f2.dm_in1k256 (prueba 352)85.1843521000apache-2.0timm
dm_nfnet_f3.dm_in1k320 (prueba 416)85.6764161000apache-2.0timm
dm_nfnet_f4.dm_in1k384 (prueba 512)85.8385121000apache-2.0timm
dm_nfnet_f5.dm_in1k416 (prueba 544)86.1025441000apache-2.0timm
dm_nfnet_f6.dm_in1k448 (prueba 576)86.3645761000apache-2.0timm

etiquetas ra2

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
eca_nfnet_l0.ra2_in1k224 (prueba 288)82.5902881000apache-2.0timm
eca_nfnet_l1.ra2_in1k256 (prueba 320)84.0623201000apache-2.0timm
nf_regnet_b1.ra2_in1k256 (prueba 288)79.3682881000apache-2.0timm
nf_resnet50.ra2_in1k256 (prueba 288)80.6582881000apache-2.0timm
nfnet_l0.ra2_in1k224 (prueba 288)82.7662881000apache-2.0timm

Otros

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
eca_nfnet_l2.ra3_in1k320 (prueba 384)84.7103841000apache-2.0timm
test_nfnet.r160_in1k16047.6341601000apache-2.0timm
Update on GitHub