timm documentation
NesT
NesT
Categoria: ViT · Código: timm/models/nest.py · Arquiteturas: 6 · Pesos pré-treinados: 3 (3 com resultados no ImageNet-1k)
Os Transformers Aninhados (NesT) aplicam autoatenção localmente dentro de blocos de imagem não sobrepostos e agregam blocos vizinhos entre os estágios. Pesos convertidos da liberação oficial em JAX.
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('nest_tiny_jx.goog_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 (Nest com nest_tiny 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 |
|---|---|---|---|---|
nest_base | 67.72 | 17.96 | 224 | 0 |
nest_base_jx | 67.72 | 17.96 | 224 | 1 |
nest_small | 38.35 | 10.35 | 224 | 0 |
nest_small_jx | 38.35 | 10.35 | 224 | 1 |
nest_tiny | 17.06 | 5.83 | 224 | 0 |
nest_tiny_jx | 17.06 | 5.83 | 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.
| Peso | Tamanho da entrada | Top-1 (%) | Tamanho da avaliação | Classes | Licença | Hub |
|---|---|---|---|---|---|---|
nest_base_jx.goog_in1k | 224 | 83.552 | 224 | 1000 | apache-2.0 | timm |
nest_small_jx.goog_in1k | 224 | 83.132 | 224 | 1000 | apache-2.0 | timm |
nest_tiny_jx.goog_in1k | 224 | 81.428 | 224 | 1000 | apache-2.0 | timm |