timm documentation

RepGhost

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

and get access to the augmented documentation experience

to get started

RepGhost

Categoria: Eficiente/Móvel · Código: timm/models/repghost.py · Arquiteturas: 8 · Pesos pré-treinados: 8 (8 com resultados no ImageNet-1k)

RepGhostNet substitui a concatenação de recursos do módulo Ghost pela reutilização de recursos reparametrizada, incorporada em um único caminho de convolução na inferência.

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

Suporte à API de modelos

RecursoRepGhostNet
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 (RepGhostNet com repghostnet_050 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
repghostnet_0502.310.052241
repghostnet_0582.540.062241
repghostnet_0803.270.102241
repghostnet_1004.060.152241
repghostnet_1114.520.182241
repghostnet_1305.460.242241
repghostnet_1506.550.312241
repghostnet_2009.770.532241

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
repghostnet_050.in1k22466.9782241000mittimm
repghostnet_058.in1k22468.9382241000mittimm
repghostnet_080.in1k22472.2402241000mittimm
repghostnet_100.in1k22474.2082241000mittimm
repghostnet_111.in1k22475.0602241000mittimm
repghostnet_130.in1k22476.3782241000mittimm
repghostnet_150.in1k22477.4742241000mittimm
repghostnet_200.in1k22478.8022241000mittimm
Update on GitHub