timm documentation

CSPNet (CSPResNet, CSPDarkNet, DarkNet)

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

CSPNet (CSPResNet, CSPDarkNet, DarkNet)

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

Las redes Cross Stage Partial dividen las características de cada etapa en dos partes, una que pasa por los bloques de la etapa y otra que se fusiona al final de la etapa. Este módulo contiene CSPResNet, CSPResNeXt, CSPDarkNet, DarkNet y las variantes timm cs3*.

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

Soporte de la API de modelos

CaracterísticaCspNet
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 (CspNet con cs3darknet_focus_s en 256), 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
cs3darknet_focus_l21.154.662561
cs3darknet_focus_m9.301.982561
cs3darknet_focus_s3.270.692561
cs3darknet_focus_x35.028.032560
cs3darknet_l21.164.862561
cs3darknet_m9.312.082561
cs3darknet_s3.280.722560
cs3darknet_x35.058.382561
cs3edgenet_x47.8211.532561
cs3se_edgenet_x50.7211.532561
cs3sedarknet_l21.914.862561
cs3sedarknet_x35.408.382561
cs3sedarknet_xdw21.605.972560
cspdarknet5327.646.572561
cspresnet5021.624.542561
cspresnet50d21.644.862560
cspresnet50w28.125.042560
cspresnext5020.574.052561
darknet1714.303.262560
darknet2120.863.932560
darknet5341.619.312561
darknetaa5336.027.972561
sedarknet2120.953.932560

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.

c2ns etiquetas

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
cs3darknet_focus_l.c2ns_in1k256 (prueba 288)80.8942881000apache-2.0timm
cs3darknet_focus_m.c2ns_in1k256 (prueba 288)77.2862881000apache-2.0timm
cs3darknet_l.c2ns_in1k256 (prueba 288)80.8822881000apache-2.0timm
cs3darknet_m.c2ns_in1k256 (prueba 288)77.6362881000apache-2.0timm
cs3darknet_x.c2ns_in1k256 (prueba 288)82.2342881000apache-2.0timm
cs3se_edgenet_x.c2ns_in1k256 (prueba 320)83.5503201000apache-2.0timm
cs3sedarknet_l.c2ns_in1k256 (prueba 288)81.7702881000apache-2.0timm
cs3sedarknet_x.c2ns_in1k256 (prueba 288)82.6542881000apache-2.0timm
darknet53.c2ns_in1k256 (prueba 288)80.5422881000apache-2.0timm
darknetaa53.c2ns_in1k256 (prueba 288)80.5282881000apache-2.0timm

etiquetas ra

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
cspdarknet53.ra_in1k25680.0782561000apache-2.0timm
cspresnet50.ra_in1k25679.5782561000apache-2.0timm
cspresnext50.ra_in1k25680.5482561000apache-2.0timm

Otros

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
cs3darknet_focus_s.ra4_e3600_r256_in1k256 (prueba 320)73.1583201000apache-2.0timm
cs3edgenet_x.c2_in1k256 (prueba 288)82.7342881000apache-2.0timm
Update on GitHub