timm documentation

FasterNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

FasterNet

Categoría: Eficiente/Móvil · Código: timm/models/fasternet.py · Arquitecturas: 6 · Pesos preentrenados: 6 (6 con resultados en ImageNet-1k)

FasterNet se basa en la convolución parcial (PConv), que convoluciona solo una parte de los canales de entrada para reducir el cálculo y el acceso a la memoria redundantes.

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

Soporte de la API de modelos

CaracterísticaFasterNet
mapas de características de features_only=True✓ (pasos 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 (FasterNet con fasternet_t0 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

ArquitecturaParámetros (M)GMACsTamaño de entradaPesos preentrenados
fasternet_l93.4715.522241
fasternet_m53.528.742241
fasternet_s31.184.562241
fasternet_t03.910.342241
fasternet_t17.600.852241
fasternet_t214.981.912241

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
fasternet_l.in1k22483.5162241000apache-2.0timm
fasternet_m.in1k22482.8482241000apache-2.0timm
fasternet_s.in1k22481.3282241000apache-2.0timm
fasternet_t0.in1k22471.7322241000apache-2.0timm
fasternet_t1.in1k22475.9222241000apache-2.0timm
fasternet_t2.in1k22478.7382241000apache-2.0timm

Cita

@article{chen2023run,
  title={Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks},
  author={Chen, Jierun and Kao, Shiu-hong and He, Hao and Zhuo, Weipeng and Wen, Song and Lee, Chul-Ho and Chan, S-H Gary},
  journal={arXiv preprint arXiv:2303.03667},
  year={2023}
}
Update on GitHub