πŸͺ Timm EncodersΒΆ

Pytorch Image Models (a.k.a. timm) has a lot of pretrained models and interface which allows using these models as encoders in smp, however, not all models are supported

  • transformer models do not have features_only functionality implemented

  • some models do not have appropriate strides

Below is a table of suitable encoders (for DeepLabV3, DeepLabV3+, and PAN dilation support is needed also)

Total number of encoders: 467

Note

To use following encoders you have to add prefix tu-, e.g. tu-adv_inception_v3

Encoder name

Support dilation

adv_inception_v3

bat_resnext26ts

βœ…

botnet26t_256

βœ…

botnet50ts_256

βœ…

cspresnet50

βœ…

cspresnet50d

βœ…

cspresnet50w

βœ…

cspresnext50

βœ…

densenet121

densenet121d

densenet161

densenet169

densenet201

densenet264

densenet264d_iabn

densenetblur121d

dla102

dla102x

dla102x2

dla169

dla34

dla46_c

dla46x_c

dla60

dla60_res2net

dla60_res2next

dla60x

dla60x_c

dm_nfnet_f0

βœ…

dm_nfnet_f1

βœ…

dm_nfnet_f2

βœ…

dm_nfnet_f3

βœ…

dm_nfnet_f4

βœ…

dm_nfnet_f5

βœ…

dm_nfnet_f6

βœ…

dpn107

dpn131

dpn68

dpn68b

dpn92

dpn98

eca_botnext26ts_256

βœ…

eca_efficientnet_b0

βœ…

eca_halonext26ts

βœ…

eca_lambda_resnext26ts

βœ…

eca_nfnet_l0

βœ…

eca_nfnet_l1

βœ…

eca_nfnet_l2

βœ…

eca_nfnet_l3

βœ…

eca_swinnext26ts_256

βœ…

eca_vovnet39b

ecaresnet101d

βœ…

ecaresnet101d_pruned

βœ…

ecaresnet200d

βœ…

ecaresnet269d

βœ…

ecaresnet26t

βœ…

ecaresnet50d

βœ…

ecaresnet50d_pruned

βœ…

ecaresnet50t

βœ…

ecaresnetlight

βœ…

ecaresnext26t_32x4d

βœ…

ecaresnext50t_32x4d

βœ…

efficientnet_b0

βœ…

efficientnet_b1

βœ…

efficientnet_b1_pruned

βœ…

efficientnet_b2

βœ…

efficientnet_b2_pruned

βœ…

efficientnet_b2a

βœ…

efficientnet_b3

βœ…

efficientnet_b3_pruned

βœ…

efficientnet_b3a

βœ…

efficientnet_b4

βœ…

efficientnet_b5

βœ…

efficientnet_b6

βœ…

efficientnet_b7

βœ…

efficientnet_b8

βœ…

efficientnet_cc_b0_4e

βœ…

efficientnet_cc_b0_8e

βœ…

efficientnet_cc_b1_8e

βœ…

efficientnet_el

βœ…

efficientnet_el_pruned

βœ…

efficientnet_em

βœ…

efficientnet_es

βœ…

efficientnet_es_pruned

βœ…

efficientnet_l2

βœ…

efficientnet_lite0

βœ…

efficientnet_lite1

βœ…

efficientnet_lite2

βœ…

efficientnet_lite3

βœ…

efficientnet_lite4

βœ…

efficientnetv2_l

βœ…

efficientnetv2_m

βœ…

efficientnetv2_rw_m

βœ…

efficientnetv2_rw_s

βœ…

efficientnetv2_s

βœ…

ens_adv_inception_resnet_v2

ese_vovnet19b_dw

ese_vovnet19b_slim

ese_vovnet19b_slim_dw

ese_vovnet39b

ese_vovnet39b_evos

ese_vovnet57b

ese_vovnet99b

ese_vovnet99b_iabn

fbnetc_100

βœ…

fbnetv3_b

βœ…

fbnetv3_d

βœ…

fbnetv3_g

βœ…

gc_efficientnet_b0

βœ…

gcresnet50t

βœ…

gcresnext26ts

βœ…

geresnet50t

βœ…

gernet_l

βœ…

gernet_m

βœ…

gernet_s

βœ…

ghostnet_050

ghostnet_100

ghostnet_130

gluon_inception_v3

gluon_resnet101_v1b

βœ…

gluon_resnet101_v1c

βœ…

gluon_resnet101_v1d

βœ…

gluon_resnet101_v1s

βœ…

gluon_resnet152_v1b

βœ…

gluon_resnet152_v1c

βœ…

gluon_resnet152_v1d

βœ…

gluon_resnet152_v1s

βœ…

gluon_resnet18_v1b

βœ…

gluon_resnet34_v1b

βœ…

gluon_resnet50_v1b

βœ…

gluon_resnet50_v1c

βœ…

gluon_resnet50_v1d

βœ…

gluon_resnet50_v1s

βœ…

gluon_resnext101_32x4d

βœ…

gluon_resnext101_64x4d

βœ…

gluon_resnext50_32x4d

βœ…

gluon_senet154

βœ…

gluon_seresnext101_32x4d

βœ…

gluon_seresnext101_64x4d

βœ…

gluon_seresnext50_32x4d

βœ…

gluon_xception65

βœ…

halonet26t

βœ…

halonet50ts

βœ…

halonet_h1

βœ…

halonet_h1_c4c5

βœ…

hardcorenas_a

βœ…

hardcorenas_b

βœ…

hardcorenas_c

βœ…

hardcorenas_d

βœ…

hardcorenas_e

βœ…

hardcorenas_f

βœ…

hrnet_w18

hrnet_w18_small

hrnet_w18_small_v2

hrnet_w30

hrnet_w32

hrnet_w40

hrnet_w44

hrnet_w48

hrnet_w64

ig_resnext101_32x16d

βœ…

ig_resnext101_32x32d

βœ…

ig_resnext101_32x48d

βœ…

ig_resnext101_32x8d

βœ…

inception_resnet_v2

inception_v3

inception_v4

lambda_resnet26t

βœ…

lambda_resnet50t

βœ…

legacy_senet154

legacy_seresnet101

legacy_seresnet152

legacy_seresnet18

legacy_seresnet34

legacy_seresnet50

legacy_seresnext101_32x4d

legacy_seresnext26_32x4d

legacy_seresnext50_32x4d

mixnet_l

βœ…

mixnet_m

βœ…

mixnet_s

βœ…

mixnet_xl

βœ…

mixnet_xxl

βœ…

mnasnet_050

βœ…

mnasnet_075

βœ…

mnasnet_100

βœ…

mnasnet_140

βœ…

mnasnet_a1

βœ…

mnasnet_b1

βœ…

mnasnet_small

βœ…

mobilenetv2_100

βœ…

mobilenetv2_110d

βœ…

mobilenetv2_120d

βœ…

mobilenetv2_140

βœ…

mobilenetv3_large_075

βœ…

mobilenetv3_large_100

βœ…

mobilenetv3_large_100_miil

βœ…

mobilenetv3_large_100_miil_in21k

βœ…

mobilenetv3_rw

βœ…

mobilenetv3_small_075

βœ…

mobilenetv3_small_100

βœ…

nasnetalarge

nf_ecaresnet101

βœ…

nf_ecaresnet26

βœ…

nf_ecaresnet50

βœ…

nf_regnet_b0

βœ…

nf_regnet_b1

βœ…

nf_regnet_b2

βœ…

nf_regnet_b3

βœ…

nf_regnet_b4

βœ…

nf_regnet_b5

βœ…

nf_resnet101

βœ…

nf_resnet26

βœ…

nf_resnet50

βœ…

nf_seresnet101

βœ…

nf_seresnet26

βœ…

nf_seresnet50

βœ…

nfnet_f0

βœ…

nfnet_f0s

βœ…

nfnet_f1

βœ…

nfnet_f1s

βœ…

nfnet_f2

βœ…

nfnet_f2s

βœ…

nfnet_f3

βœ…

nfnet_f3s

βœ…

nfnet_f4

βœ…

nfnet_f4s

βœ…

nfnet_f5

βœ…

nfnet_f5s

βœ…

nfnet_f6

βœ…

nfnet_f6s

βœ…

nfnet_f7

βœ…

nfnet_f7s

βœ…

nfnet_l0

βœ…

pnasnet5large

rednet26t

βœ…

rednet50ts

βœ…

regnetx_002

βœ…

regnetx_004

βœ…

regnetx_006

βœ…

regnetx_008

βœ…

regnetx_016

βœ…

regnetx_032

βœ…

regnetx_040

βœ…

regnetx_064

βœ…

regnetx_080

βœ…

regnetx_120

βœ…

regnetx_160

βœ…

regnetx_320

βœ…

regnety_002

βœ…

regnety_004

βœ…

regnety_006

βœ…

regnety_008

βœ…

regnety_016

βœ…

regnety_032

βœ…

regnety_040

βœ…

regnety_064

βœ…

regnety_080

βœ…

regnety_120

βœ…

regnety_160

βœ…

regnety_320

βœ…

repvgg_a2

βœ…

repvgg_b0

βœ…

repvgg_b1

βœ…

repvgg_b1g4

βœ…

repvgg_b2

βœ…

repvgg_b2g4

βœ…

repvgg_b3

βœ…

repvgg_b3g4

βœ…

res2net101_26w_4s

βœ…

res2net50_14w_8s

βœ…

res2net50_26w_4s

βœ…

res2net50_26w_6s

βœ…

res2net50_26w_8s

βœ…

res2net50_48w_2s

βœ…

res2next50

βœ…

resnest101e

βœ…

resnest14d

βœ…

resnest200e

βœ…

resnest269e

βœ…

resnest26d

βœ…

resnest50d

βœ…

resnest50d_1s4x24d

βœ…

resnest50d_4s2x40d

βœ…

resnet101

βœ…

resnet101d

βœ…

resnet152

βœ…

resnet152d

βœ…

resnet18

βœ…

resnet18d

βœ…

resnet200

βœ…

resnet200d

βœ…

resnet26

βœ…

resnet26d

βœ…

resnet26t

βœ…

resnet34

βœ…

resnet34d

βœ…

resnet50

βœ…

resnet50d

βœ…

resnet50t

βœ…

resnet51q

βœ…

resnet61q

βœ…

resnetblur18

βœ…

resnetblur50

βœ…

resnetrs101

βœ…

resnetrs152

βœ…

resnetrs200

βœ…

resnetrs270

βœ…

resnetrs350

βœ…

resnetrs420

βœ…

resnetrs50

βœ…

resnetv2_101

βœ…

resnetv2_101d

βœ…

resnetv2_101x1_bitm

βœ…

resnetv2_101x1_bitm_in21k

βœ…

resnetv2_101x3_bitm

βœ…

resnetv2_101x3_bitm_in21k

βœ…

resnetv2_152

βœ…

resnetv2_152d

βœ…

resnetv2_152x2_bit_teacher

βœ…

resnetv2_152x2_bit_teacher_384

βœ…

resnetv2_152x2_bitm

βœ…

resnetv2_152x2_bitm_in21k

βœ…

resnetv2_152x4_bitm

βœ…

resnetv2_152x4_bitm_in21k

βœ…

resnetv2_50

βœ…

resnetv2_50d

βœ…

resnetv2_50t

βœ…

resnetv2_50x1_bit_distilled

βœ…

resnetv2_50x1_bitm

βœ…

resnetv2_50x1_bitm_in21k

βœ…

resnetv2_50x3_bitm

βœ…

resnetv2_50x3_bitm_in21k

βœ…

resnext101_32x4d

βœ…

resnext101_32x8d

βœ…

resnext101_64x4d

βœ…

resnext50_32x4d

βœ…

resnext50d_32x4d

βœ…

rexnet_100

rexnet_130

rexnet_150

rexnet_200

rexnetr_100

rexnetr_130

rexnetr_150

rexnetr_200

selecsls42

selecsls42b

selecsls60

selecsls60b

selecsls84

semnasnet_050

βœ…

semnasnet_075

βœ…

semnasnet_100

βœ…

semnasnet_140

βœ…

senet154

βœ…

seresnet101

βœ…

seresnet152

βœ…

seresnet152d

βœ…

seresnet18

βœ…

seresnet200d

βœ…

seresnet269d

βœ…

seresnet34

βœ…

seresnet50

βœ…

seresnet50t

βœ…

seresnext101_32x4d

βœ…

seresnext101_32x8d

βœ…

seresnext26d_32x4d

βœ…

seresnext26t_32x4d

βœ…

seresnext26tn_32x4d

βœ…

seresnext50_32x4d

βœ…

skresnet18

βœ…

skresnet34

βœ…

skresnet50

βœ…

skresnet50d

βœ…

skresnext50_32x4d

βœ…

spnasnet_100

βœ…

ssl_resnet18

βœ…

ssl_resnet50

βœ…

ssl_resnext101_32x16d

βœ…

ssl_resnext101_32x4d

βœ…

ssl_resnext101_32x8d

βœ…

ssl_resnext50_32x4d

βœ…

swinnet26t_256

βœ…

swinnet50ts_256

βœ…

swsl_resnet18

βœ…

swsl_resnet50

βœ…

swsl_resnext101_32x16d

βœ…

swsl_resnext101_32x4d

βœ…

swsl_resnext101_32x8d

βœ…

swsl_resnext50_32x4d

βœ…

tf_efficientnet_b0

βœ…

tf_efficientnet_b0_ap

βœ…

tf_efficientnet_b0_ns

βœ…

tf_efficientnet_b1

βœ…

tf_efficientnet_b1_ap

βœ…

tf_efficientnet_b1_ns

βœ…

tf_efficientnet_b2

βœ…

tf_efficientnet_b2_ap

βœ…

tf_efficientnet_b2_ns

βœ…

tf_efficientnet_b3

βœ…

tf_efficientnet_b3_ap

βœ…

tf_efficientnet_b3_ns

βœ…

tf_efficientnet_b4

βœ…

tf_efficientnet_b4_ap

βœ…

tf_efficientnet_b4_ns

βœ…

tf_efficientnet_b5

βœ…

tf_efficientnet_b5_ap

βœ…

tf_efficientnet_b5_ns

βœ…

tf_efficientnet_b6

βœ…

tf_efficientnet_b6_ap

βœ…

tf_efficientnet_b6_ns

βœ…

tf_efficientnet_b7

βœ…

tf_efficientnet_b7_ap

βœ…

tf_efficientnet_b7_ns

βœ…

tf_efficientnet_b8

βœ…

tf_efficientnet_b8_ap

βœ…

tf_efficientnet_cc_b0_4e

βœ…

tf_efficientnet_cc_b0_8e

βœ…

tf_efficientnet_cc_b1_8e

βœ…

tf_efficientnet_el

βœ…

tf_efficientnet_em

βœ…

tf_efficientnet_es

βœ…

tf_efficientnet_l2_ns

βœ…

tf_efficientnet_l2_ns_475

βœ…

tf_efficientnet_lite0

βœ…

tf_efficientnet_lite1

βœ…

tf_efficientnet_lite2

βœ…

tf_efficientnet_lite3

βœ…

tf_efficientnet_lite4

βœ…

tf_efficientnetv2_b0

βœ…

tf_efficientnetv2_b1

βœ…

tf_efficientnetv2_b2

βœ…

tf_efficientnetv2_b3

βœ…

tf_efficientnetv2_l

βœ…

tf_efficientnetv2_l_in21ft1k

βœ…

tf_efficientnetv2_l_in21k

βœ…

tf_efficientnetv2_m

βœ…

tf_efficientnetv2_m_in21ft1k

βœ…

tf_efficientnetv2_m_in21k

βœ…

tf_efficientnetv2_s

βœ…

tf_efficientnetv2_s_in21ft1k

βœ…

tf_efficientnetv2_s_in21k

βœ…

tf_inception_v3

tf_mixnet_l

βœ…

tf_mixnet_m

βœ…

tf_mixnet_s

βœ…

tf_mobilenetv3_large_075

βœ…

tf_mobilenetv3_large_100

βœ…

tf_mobilenetv3_large_minimal_100

βœ…

tf_mobilenetv3_small_075

βœ…

tf_mobilenetv3_small_100

βœ…

tf_mobilenetv3_small_minimal_100

βœ…

tv_densenet121

tv_resnet101

βœ…

tv_resnet152

βœ…

tv_resnet34

βœ…

tv_resnet50

βœ…

tv_resnext50_32x4d

βœ…

vovnet39a

vovnet57a

wide_resnet101_2

βœ…

wide_resnet50_2

βœ…

xception

xception41

βœ…

xception65

βœ…

xception71

βœ…