timm documentation

DenseNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

DenseNet

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

Redes convolucionais densamente conectadas, em que cada camada em um bloco denso recebe as saídas concatenadas de todas as camadas anteriores do bloco. Adaptado do torchvision, densenetblur usa downsampling anti-aliasado com blur pool.

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

Suporte à API de modelos

RecursoDenseNet
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 (DenseNet com densenet121 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
densenet1217.982.872242
densenet16128.687.792241
densenet16914.153.402241
densenet20120.014.342241
densenet264d72.7413.572240
densenetblur121d8.003.112241

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 tv

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
densenet121.tv_in1k22474.7502241000apache-2.0timm
densenet161.tv_in1k22477.3842241000apache-2.0timm
densenet169.tv_in1k22475.9222241000apache-2.0timm
densenet201.tv_in1k22477.2842241000apache-2.0timm

ra tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
densenet121.ra_in1k224 (teste 288)76.4862881000apache-2.0timm
densenetblur121d.ra_in1k224 (teste 288)77.3202881000apache-2.0timm
Update on GitHub