timm documentation

StarNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

StarNet

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

StarNet es una red de concepto demostrativo deliberadamente simple construida en torno a la operación de estrella, la multiplicación elemento a elemento de dos 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('starnet_s4.in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Soporte de la API de modelos

CaracterísticaStarNet
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 (StarNet con starnet_s050 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
starnet_s12.870.422241
starnet_s23.680.552241
starnet_s35.750.762241
starnet_s47.481.052241
starnet_s0500.540.092240
starnet_s1001.040.192240
starnet_s1501.560.232240

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
starnet_s1.in1k22473.5322241000apache-2.0timm
starnet_s2.in1k22474.6662241000apache-2.0timm
starnet_s3.in1k22477.3782241000apache-2.0timm
starnet_s4.in1k22478.8242241000apache-2.0timm
Update on GitHub