timm documentation

HRNet

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.31).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

HRNet

Categoria: CNN · Código: timm/models/hrnet.py · Arquiteturas: 11 · Pesos pré-treinados: 14 (14 com resultados no ImageNet-1k)

As Redes de Alta Resolução mantêm um ramo de alta resolução por toda a rede e adicionam ramos de resolução mais baixa em paralelo, trocando informações entre eles repetidamente.

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

Suporte à API de modelos

RecursoHighResolutionNet
mapas de recursos de features_only=True✓ (passos 2, 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 (HighResolutionNet com hrnet_w18_small 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
hrnet_w1821.304.322242
hrnet_w18_small13.191.612242
hrnet_w18_small_v215.602.622242
hrnet_w18_ssld21.304.322241
hrnet_w3037.718.152241
hrnet_w3241.238.972241
hrnet_w4057.5612.752241
hrnet_w4467.0614.942241
hrnet_w4877.4717.342241
hrnet_w48_ssld77.4717.342241
hrnet_w64128.0628.972241

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.

Os pesos são agrupados por origem: o artigo ou origem indicado na configuração pré-treinada, caso contrário, uma organização do Hub diferente de timm, caso contrário, a primeira parte da tag pré-treinada (architecture.tag). As origens com um único peso são listadas em other.

tags de ms

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
hrnet_w18.ms_aug_in1k22478.1102241000mittimm
hrnet_w18.ms_in1k22476.7702241000mittimm
hrnet_w18_small.ms_in1k22472.3402241000mittimm
hrnet_w18_small_v2.ms_in1k22475.0942241000mittimm
hrnet_w30.ms_in1k22478.2022241000mittimm
hrnet_w32.ms_in1k22478.4522241000mittimm
hrnet_w40.ms_in1k22478.9282241000mittimm
hrnet_w44.ms_in1k22478.8902241000mittimm
hrnet_w48.ms_in1k22479.3282241000mittimm
hrnet_w64.ms_in1k22479.4582241000mittimm

gluon tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
hrnet_w18_small.gluon_in1k22473.9342241000apache-2.0timm
hrnet_w18_small_v2.gluon_in1k22478.2022241000apache-2.0timm

paddle tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
hrnet_w18_ssld.paddle_in1k224 (teste 288)82.0522881000mittimm
hrnet_w48_ssld.paddle_in1k224 (teste 288)84.4842881000mittimm
Update on GitHub