timm documentation
HRNet
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
- Aprendizado de Representação de Alta Resolução Profunda para Reconhecimento Visual
- Além da auto-supervisão: uma alternativa simples e eficaz de destilação de rede para melhorar os backbones
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
✓ 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
| Arquitetura | Parâmetros (M) | GMACs | Tamanho da entrada | Pesos pré-treinados |
|---|---|---|---|---|
hrnet_w18 | 21.30 | 4.32 | 224 | 2 |
hrnet_w18_small | 13.19 | 1.61 | 224 | 2 |
hrnet_w18_small_v2 | 15.60 | 2.62 | 224 | 2 |
hrnet_w18_ssld | 21.30 | 4.32 | 224 | 1 |
hrnet_w30 | 37.71 | 8.15 | 224 | 1 |
hrnet_w32 | 41.23 | 8.97 | 224 | 1 |
hrnet_w40 | 57.56 | 12.75 | 224 | 1 |
hrnet_w44 | 67.06 | 14.94 | 224 | 1 |
hrnet_w48 | 77.47 | 17.34 | 224 | 1 |
hrnet_w48_ssld | 77.47 | 17.34 | 224 | 1 |
hrnet_w64 | 128.06 | 28.97 | 224 | 1 |
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
| Peso | Tamanho da entrada | Top-1 (%) | Tamanho da avaliação | Classes | Licença | Hub |
|---|---|---|---|---|---|---|
hrnet_w18.ms_aug_in1k | 224 | 78.110 | 224 | 1000 | mit | timm |
hrnet_w18.ms_in1k | 224 | 76.770 | 224 | 1000 | mit | timm |
hrnet_w18_small.ms_in1k | 224 | 72.340 | 224 | 1000 | mit | timm |
hrnet_w18_small_v2.ms_in1k | 224 | 75.094 | 224 | 1000 | mit | timm |
hrnet_w30.ms_in1k | 224 | 78.202 | 224 | 1000 | mit | timm |
hrnet_w32.ms_in1k | 224 | 78.452 | 224 | 1000 | mit | timm |
hrnet_w40.ms_in1k | 224 | 78.928 | 224 | 1000 | mit | timm |
hrnet_w44.ms_in1k | 224 | 78.890 | 224 | 1000 | mit | timm |
hrnet_w48.ms_in1k | 224 | 79.328 | 224 | 1000 | mit | timm |
hrnet_w64.ms_in1k | 224 | 79.458 | 224 | 1000 | mit | timm |
gluon tags
| Peso | Tamanho da entrada | Top-1 (%) | Tamanho da avaliação | Classes | Licença | Hub |
|---|---|---|---|---|---|---|
hrnet_w18_small.gluon_in1k | 224 | 73.934 | 224 | 1000 | apache-2.0 | timm |
hrnet_w18_small_v2.gluon_in1k | 224 | 78.202 | 224 | 1000 | apache-2.0 | timm |
paddle tags
| Peso | Tamanho da entrada | Top-1 (%) | Tamanho da avaliação | Classes | Licença | Hub |
|---|---|---|---|---|---|---|
hrnet_w18_ssld.paddle_in1k | 224 (teste 288) | 82.052 | 288 | 1000 | mit | timm |
hrnet_w48_ssld.paddle_in1k | 224 (teste 288) | 84.484 | 288 | 1000 | mit | timm |