timm documentation

FocalNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

FocalNet

Categoria: Híbrido · Código: timm/models/focalnet.py · Arquiteturas: 12 · Pesos pré-treinados: 12 (6 com resultados no ImageNet-1k)

Focal Modulation Networks (FocalNet) substituem a autoatenção pela focal modulation, que agrega contexto em várias granularidades com convoluções depthwise e o direciona para cada token de consulta.

Artigos

Uso

Veja o Início rápido para carregar pesos pré-treinados, pré-processar imagens e fazer fine-tuning, e Extração de recursos para usar os modelos como backbones.

>>> import timm
>>> model = timm.create_model('focalnet_tiny_srf.ms_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Suporte à API de modelos

RecursoFocalNet
mapas de recursos de features_only=True✓ (passos 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Qualquer tamanho de entrada, sem necessidade de alterações✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Decaimento de LR por camada (group_matcher())✓
reset_classifier()✓

✓ suportado, ✓ (quadrado) funciona apenas para tamanhos quadrados, ✗ não suportado, – não necessário (o modelo já aceita qualquer tamanho de entrada). Verificado executando cada recurso em um modelo inicializado aleatoriamente (FocalNet com focalnet_tiny_srf em 224), espera-se que outras arquiteturas da mesma classe se comportem da mesma forma. Consulte a página Model API para mais detalhes.

Arquiteturas

ArquiteturaParâmetros (M)GMACsTamanho da entradaPesos pré-treinados
focalnet_base_lrf88.7515.432241
focalnet_base_srf88.1515.282241
focalnet_huge_fl3745.28118.262241
focalnet_huge_fl4686.46118.902241
focalnet_large_fl3239.13105.063841
focalnet_large_fl4239.32105.203841
focalnet_small_lrf50.348.742241
focalnet_small_srf49.898.622241
focalnet_tiny_lrf28.654.492241
focalnet_tiny_srf28.434.422241
focalnet_xlarge_fl3408.79185.613841
focalnet_xlarge_fl4409.03185.793841

Os GMACs e as contagens de parâmetros são para o tamanho de entrada e o classificador padrão de cada arquitetura, dos resultados do benchmark.

Pesos pré-treinados

Top-1 é a acurácia de validação no ImageNet-1k de results-imagenet.csv no tamanho de avaliação listado, - se o peso não tiver resultado no ImageNet-1k (por exemplo, pesos de pré-treinamento ou de imagem-texto). Classes é o tamanho do classificador pré-treinado, nenhum para pesos sem um.

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
focalnet_base_lrf.ms_in1k22483.8282241000mittimm
focalnet_base_srf.ms_in1k22483.8082241000mittimm
focalnet_huge_fl3.ms_in22k224--21842mittimm
focalnet_huge_fl4.ms_in22k224--nonemittimm
focalnet_large_fl3.ms_in22k384--21842mittimm
focalnet_large_fl4.ms_in22k384--21842mittimm
focalnet_small_lrf.ms_in1k22483.5082241000mittimm
focalnet_small_srf.ms_in1k22483.4282241000mittimm
focalnet_tiny_lrf.ms_in1k22482.1742241000mittimm
focalnet_tiny_srf.ms_in1k22482.1462241000mittimm
focalnet_xlarge_fl3.ms_in22k384--21842mittimm
focalnet_xlarge_fl4.ms_in22k384--21842mittimm
Update on GitHub