timm documentation

EfficientNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

EfficientNet

Categoria: Efficient/Mobile · Código: timm/models/efficientnet.py · Arquiteturas: 115 · Pesos pré-treinados: 119 (113 com resultados no ImageNet-1k)

EfficientNet e EfficientNetV2, e as arquiteturas móveis relacionadas construídas a partir dos mesmos blocos de separável em profundidade, residual invertido (MBConv) e MBConv fundido: MixNet, MnasNet, FBNet, Single-Path NAS, MobileNet-V1 / V2, EfficientNet-Lite, EdgeTPU e TinyNet.

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

Suporte à API de modelos

RecursoEfficientNet
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 (EfficientNet com mobilenetv2_035 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
efficientnet_b05.290.402242
efficientnet_b0_g8_gn6.560.662240
efficientnet_b0_g16_evos8.111.012240
efficientnet_b0_gn5.290.422240
efficientnet_b17.790.712402
efficientnet_b1_pruned6.330.402401
efficientnet_b29.110.892561
efficientnet_b2_pruned8.310.732601
efficientnet_b312.231.632881
efficientnet_b3_g8_gn14.252.592880
efficientnet_b3_gn11.731.742880
efficientnet_b3_pruned9.861.043001
efficientnet_b419.343.133201
efficientnet_b530.399.594482
efficientnet_b643.0419.405280
efficientnet_b766.3538.336000
efficientnet_b887.4163.486720
efficientnet_blur_b05.290.432240
efficientnet_cc_b0_4e13.310.412240
efficientnet_cc_b0_8e24.010.422240
efficientnet_cc_b1_8e39.720.752400
efficientnet_el10.598.003001
efficientnet_el_pruned10.598.003001
efficientnet_em6.903.042401
efficientnet_es5.441.812241
efficientnet_es_pruned5.441.812241
efficientnet_h_b545.8827.164481
efficientnet_l2480.31479.128000
efficientnet_lite04.650.402241
efficientnet_lite15.420.622400
efficientnet_lite26.090.892600
efficientnet_lite38.201.653000
efficientnet_lite413.014.043800
efficientnet_x_b313.303.912880
efficientnet_x_b533.4423.354481
efficientnetv2_l118.5236.103840
efficientnetv2_m54.1411.013200
efficientnetv2_rw_m53.2412.723201
efficientnetv2_rw_s23.944.912881
efficientnetv2_rw_t13.651.932241
efficientnetv2_s21.464.752880
efficientnetv2_xl208.1252.813840
fbnetc_1005.570.402241
gc_efficientnetv2_rw_t13.681.942241
mixnet_l7.330.582241
mixnet_m5.010.362241
mixnet_s4.130.252241
mixnet_xl11.900.932241
mixnet_xxl23.962.042240
mnasnet_0502.220.112240
mnasnet_0753.170.232240
mnasnet_1004.380.332241
mnasnet_1407.120.602240
mnasnet_small2.030.072241
mobilenet_edgetpu_1004.091.002240
mobilenet_edgetpu_v2_l10.922.552240
mobilenet_edgetpu_v2_m8.461.852241
mobilenet_edgetpu_v2_s5.991.212240
mobilenet_edgetpu_v2_xs4.460.702240
mobilenetv1_1004.230.582241
mobilenetv1_100h5.280.632241
mobilenetv1_1256.270.892241
mobilenetv2_0351.680.072240
mobilenetv2_0501.970.102241
mobilenetv2_0752.640.222240
mobilenetv2_1003.500.312241
mobilenetv2_110d4.520.452241
mobilenetv2_120d5.830.692241
mobilenetv2_1406.110.602241
semnasnet_0502.080.112240
semnasnet_0752.910.232241
semnasnet_1003.890.322241
semnasnet_1406.110.602240
spnasnet_1004.420.352241
test_efficientnet0.360.061601
test_efficientnet_evos0.360.061601
test_efficientnet_gn0.360.061601
test_efficientnet_ln0.360.061601
tf_efficientnet_b05.290.402244
tf_efficientnet_b17.790.712404
tf_efficientnet_b29.111.022604
tf_efficientnet_b312.231.873004
tf_efficientnet_b419.344.493804
tf_efficientnet_b530.3910.464565
tf_efficientnet_b643.0419.405283
tf_efficientnet_b766.3538.336004
tf_efficientnet_b887.4163.486722
tf_efficientnet_cc_b0_4e13.310.412241
tf_efficientnet_cc_b0_8e24.010.422241
tf_efficientnet_cc_b1_8e39.720.752401
tf_efficientnet_el10.598.003001
tf_efficientnet_em6.903.042401
tf_efficientnet_es5.441.812241
tf_efficientnet_l2480.31172.114752
tf_efficientnet_lite04.650.402241
tf_efficientnet_lite15.420.622401
tf_efficientnet_lite26.090.892601
tf_efficientnet_lite38.201.653001
tf_efficientnet_lite413.014.043801
tf_efficientnetv2_b07.140.541921
tf_efficientnetv2_b18.140.761921
tf_efficientnetv2_b210.101.062081
tf_efficientnetv2_b314.361.932403
tf_efficientnetv2_l118.5236.103843
tf_efficientnetv2_m54.1415.853843
tf_efficientnetv2_s21.465.353003
tf_efficientnetv2_xl208.1252.813842
tf_mixnet_l7.330.582241
tf_mixnet_m5.010.362241
tf_mixnet_s4.130.252241
tinynet_a6.190.351921
tinynet_b3.730.211881
tinynet_c2.460.111841
tinynet_d2.340.051521
tinynet_e2.040.031061

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.

ra tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
efficientnet_b0.ra_in1k22477.6982241000apache-2.0timm
efficientnet_b2.ra_in1k256 (teste 288)80.6122881000apache-2.0timm
efficientnet_el.ra_in1k30081.3083001000apache-2.0timm
efficientnet_es.ra_in1k22478.1002241000apache-2.0timm
efficientnet_lite0.ra_in1k22475.4882241000apache-2.0timm
mixnet_xl.ra_in1k22480.4942241000apache-2.0timm
mobilenetv2_100.ra_in1k22472.9102241000apache-2.0timm
mobilenetv2_110d.ra_in1k22475.0562241000apache-2.0timm
mobilenetv2_120d.ra_in1k22477.3002241000apache-2.0timm
mobilenetv2_140.ra_in1k22476.5282241000apache-2.0timm
tf_efficientnet_b5.ra_in1k45683.8344561000apache-2.0timm
tf_efficientnet_b7.ra_in1k60084.9266001000apache-2.0timm
tf_efficientnet_b8.ra_in1k67285.3446721000apache-2.0timm

ns tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
tf_efficientnet_b0.ns_jft_in1k22478.6802241000apache-2.0timm
tf_efficientnet_b1.ns_jft_in1k24081.3882401000apache-2.0timm
tf_efficientnet_b2.ns_jft_in1k26082.4122601000apache-2.0timm
tf_efficientnet_b3.ns_jft_in1k30084.0543001000apache-2.0timm
tf_efficientnet_b4.ns_jft_in1k38085.1563801000apache-2.0timm
tf_efficientnet_b5.ns_jft_in1k45686.0964561000apache-2.0timm
tf_efficientnet_b6.ns_jft_in1k52886.4765281000apache-2.0timm
tf_efficientnet_b7.ns_jft_in1k60086.8506001000apache-2.0timm
tf_efficientnet_l2.ns_jft_in1k80088.3608001000apache-2.0timm
tf_efficientnet_l2.ns_jft_in1k_47547588.2464751000apache-2.0timm

ap tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
tf_efficientnet_b0.ap_in1k22477.1002241000apache-2.0timm
tf_efficientnet_b1.ap_in1k24079.2882401000apache-2.0timm
tf_efficientnet_b2.ap_in1k26080.3002601000apache-2.0timm
tf_efficientnet_b3.ap_in1k30081.8423001000apache-2.0timm
tf_efficientnet_b4.ap_in1k38083.2463801000apache-2.0timm
tf_efficientnet_b5.ap_in1k45684.2584561000apache-2.0timm
tf_efficientnet_b6.ap_in1k52884.7945281000apache-2.0timm
tf_efficientnet_b7.ap_in1k60085.1326001000apache-2.0timm
tf_efficientnet_b8.ap_in1k67285.3586721000apache-2.0timm

aa tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
tf_efficientnet_b0.aa_in1k22476.8822241000apache-2.0timm
tf_efficientnet_b1.aa_in1k24078.8402401000apache-2.0timm
tf_efficientnet_b2.aa_in1k26080.0742601000apache-2.0timm
tf_efficientnet_b3.aa_in1k30081.6483001000apache-2.0timm
tf_efficientnet_b4.aa_in1k38083.0403801000apache-2.0timm
tf_efficientnet_b5.aa_in1k45683.6964561000apache-2.0timm
tf_efficientnet_b6.aa_in1k52884.1245281000apache-2.0timm
tf_efficientnet_b7.aa_in1k60084.4246001000apache-2.0timm

tags ra4

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
efficientnet_b0.ra4_e3600_r224_in1k224 (teste 256)79.3642561000apache-2.0timm
efficientnet_b1.ra4_e3600_r240_in1k240 (teste 288)81.4442881000apache-2.0timm
mobilenet_edgetpu_v2_m.ra4_e3600_r224_in1k224 (teste 256)80.6842561000apache-2.0timm
mobilenetv1_100.ra4_e3600_r224_in1k224 (teste 256)76.0882561000apache-2.0timm
mobilenetv1_100h.ra4_e3600_r224_in1k224 (teste 256)76.5982561000apache-2.0timm
mobilenetv1_125.ra4_e3600_r224_in1k224 (teste 256)77.6122561000apache-2.0timm

ra2 tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
efficientnet_b3.ra2_in1k288 (teste 320)82.2963201000apache-2.0timm
efficientnet_b4.ra2_in1k320 (teste 384)83.4683841000apache-2.0timm
efficientnet_em.ra2_in1k24079.2562401000apache-2.0timm
efficientnetv2_rw_s.ra2_in1k288 (teste 384)83.8303841000apache-2.0timm
efficientnetv2_rw_t.ra2_in1k224 (teste 288)82.3542881000apache-2.0timm

tags rmsp

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
fbnetc_100.rmsp_in1k22475.1202241000apache-2.0timm
mnasnet_100.rmsp_in1k22474.6722241000apache-2.0timm
semnasnet_075.rmsp_in1k22473.0002241000apache-2.0timm
semnasnet_100.rmsp_in1k22475.4522241000apache-2.0timm
spnasnet_100.rmsp_in1k22474.0862241000apache-2.0timm

sw tags

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
efficientnet_b5.sw_in12k416--11821apache-2.0timm
efficientnet_b5.sw_in12k_ft_in1k44885.8944481000apache-2.0timm
efficientnet_h_b5.sw_r448_e450_in1k448 (teste 576)84.9425761000apache-2.0timm
efficientnet_x_b5.sw_r448_e450_in1k448 (teste 576)84.9285761000apache-2.0timm

tags agc

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
efficientnetv2_rw_m.agc_in1k320 (teste 416)84.8304161000apache-2.0timm
gc_efficientnetv2_rw_t.agc_in1k224 (teste 288)82.4862881000apache-2.0timm

tags lamb

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
mnasnet_small.lamb_in1k22466.2042241000apache-2.0timm
mobilenetv2_050.lamb_in1k22465.9282241000apache-2.0timm

Outros

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
efficientnet_b1.ft_in1k224 (teste 256)78.8062561000apache-2.0timm
efficientnet_b1_pruned.in1k24078.2522401000apache-2.0timm
efficientnet_b2_pruned.in1k26079.9042601000apache-2.0timm
efficientnet_b3_pruned.in1k30080.8643001000apache-2.0timm
efficientnet_el_pruned.in1k30080.2843001000apache-2.0timm
efficientnet_es_pruned.in1k22475.0102241000apache-2.0timm
mixnet_l.ft_in1k22478.9702241000apache-2.0timm
mixnet_m.ft_in1k22477.2682241000apache-2.0timm
mixnet_s.ft_in1k22476.0002241000apache-2.0timm
test_efficientnet.r160_in1k16046.5061601000apache-2.0timm
test_efficientnet_evos.r160_in1k16046.5041601000apache-2.0timm
test_efficientnet_gn.r160_in1k16043.8881601000apache-2.0timm
test_efficientnet_ln.r160_in1k16043.9581601000apache-2.0timm
tf_efficientnet_b0.in1k22476.5422241000apache-2.0timm
tf_efficientnet_b1.in1k24078.5462401000apache-2.0timm
tf_efficientnet_b2.in1k26079.6122601000apache-2.0timm
tf_efficientnet_b3.in1k30080.8783001000apache-2.0timm
tf_efficientnet_b4.in1k38082.5983801000apache-2.0timm
tf_efficientnet_b5.in1k45683.1704561000apache-2.0timm
tf_efficientnet_cc_b0_4e.in1k22477.3242241000apache-2.0timm
tf_efficientnet_cc_b0_8e.in1k22477.9562241000apache-2.0timm
tf_efficientnet_cc_b1_8e.in1k24079.3182401000apache-2.0timm
tf_efficientnet_el.in1k30080.2463001000apache-2.0timm
tf_efficientnet_em.in1k24078.1282401000apache-2.0timm
tf_efficientnet_es.in1k22476.6082241000apache-2.0timm
tf_efficientnet_lite0.in1k22474.8462241000apache-2.0timm
tf_efficientnet_lite1.in1k24076.6742401000apache-2.0timm
tf_efficientnet_lite2.in1k26077.4702601000apache-2.0timm
tf_efficientnet_lite3.in1k30079.8083001000apache-2.0timm
tf_efficientnet_lite4.in1k38081.5403801000apache-2.0timm
tf_efficientnetv2_b0.in1k192 (teste 224)78.3862241000apache-2.0timm
tf_efficientnetv2_b1.in1k192 (teste 240)79.5182401000apache-2.0timm
tf_efficientnetv2_b2.in1k208 (teste 260)80.2262601000apache-2.0timm
tf_efficientnetv2_b3.in1k240 (teste 300)81.9603001000apache-2.0timm
tf_efficientnetv2_b3.in21k240 (teste 300)--21843apache-2.0timm
tf_efficientnetv2_b3.in21k_ft_in1k240 (teste 300)82.6343001000apache-2.0timm
tf_efficientnetv2_l.in1k384 (teste 480)85.6604801000apache-2.0timm
tf_efficientnetv2_l.in21k384 (teste 480)--21843apache-2.0timm
tf_efficientnetv2_l.in21k_ft_in1k384 (teste 480)86.8024801000apache-2.0timm
tf_efficientnetv2_m.in1k384 (teste 480)85.2084801000apache-2.0timm
tf_efficientnetv2_m.in21k384 (teste 480)--21843apache-2.0timm
tf_efficientnetv2_m.in21k_ft_in1k384 (teste 480)86.0144801000apache-2.0timm
tf_efficientnetv2_s.in1k300 (teste 384)83.9003841000apache-2.0timm
tf_efficientnetv2_s.in21k300 (teste 384)--21843apache-2.0timm
tf_efficientnetv2_s.in21k_ft_in1k300 (teste 384)84.3323841000apache-2.0timm
tf_efficientnetv2_xl.in21k384 (teste 512)--21843apache-2.0timm
tf_efficientnetv2_xl.in21k_ft_in1k384 (teste 512)86.7545121000apache-2.0timm
tf_mixnet_l.in1k22478.7742241000apache-2.0timm
tf_mixnet_m.in1k22476.9522241000apache-2.0timm
tf_mixnet_s.in1k22475.6562241000apache-2.0timm
tinynet_a.in1k19277.6661921000apache-2.0timm
tinynet_b.in1k18874.9461881000apache-2.0timm
tinynet_c.in1k18471.2141841000apache-2.0timm
tinynet_d.in1k15266.9361521000apache-2.0timm
tinynet_e.in1k10659.8741061000apache-2.0timm
Update on GitHub