πͺ 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
not all transformer models have
features_only
functionality implemented that is required for encodersome models have inappropriate strides
Below is a table of suitable encoders (for DeepLabV3, DeepLabV3+, and PAN dilation support is needed also)
Total number of encoders: 549
Note
To use following encoders you have to add prefix tu-
, e.g. tu-adv_inception_v3
Encoder name |
Support dilation |
---|---|
SelecSls42 |
|
SelecSls42b |
|
SelecSls60 |
|
SelecSls60b |
|
SelecSls84 |
|
bat_resnext26ts |
β |
botnet26t_256 |
β |
botnet50ts_256 |
β |
coatnet_0_224 |
|
coatnet_0_rw_224 |
|
coatnet_1_224 |
|
coatnet_1_rw_224 |
|
coatnet_2_224 |
|
coatnet_2_rw_224 |
|
coatnet_3_224 |
|
coatnet_3_rw_224 |
|
coatnet_4_224 |
|
coatnet_5_224 |
|
coatnet_bn_0_rw_224 |
|
coatnet_nano_cc_224 |
|
coatnet_nano_rw_224 |
|
coatnet_pico_rw_224 |
|
coatnet_rmlp_0_rw_224 |
|
coatnet_rmlp_1_rw2_224 |
|
coatnet_rmlp_1_rw_224 |
|
coatnet_rmlp_2_rw_224 |
|
coatnet_rmlp_2_rw_384 |
|
coatnet_rmlp_3_rw_224 |
|
coatnet_rmlp_nano_rw_224 |
|
coatnext_nano_rw_224 |
|
cs3darknet_focus_l |
β |
cs3darknet_focus_m |
β |
cs3darknet_focus_s |
β |
cs3darknet_focus_x |
β |
cs3darknet_l |
β |
cs3darknet_m |
β |
cs3darknet_s |
β |
cs3darknet_x |
β |
cs3edgenet_x |
β |
cs3se_edgenet_x |
β |
cs3sedarknet_l |
β |
cs3sedarknet_x |
β |
cs3sedarknet_xdw |
β |
cspresnet50 |
β |
cspresnet50d |
β |
cspresnet50w |
β |
cspresnext50 |
β |
densenet121 |
|
densenet161 |
|
densenet169 |
|
densenet201 |
|
densenet264d |
|
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 |
|
dpn48b |
|
dpn68 |
|
dpn68b |
|
dpn92 |
|
dpn98 |
|
eca_botnext26ts_256 |
β |
eca_halonext26ts |
β |
eca_nfnet_l0 |
β |
eca_nfnet_l1 |
β |
eca_nfnet_l2 |
β |
eca_nfnet_l3 |
β |
eca_resnet33ts |
β |
eca_resnext26ts |
β |
eca_vovnet39b |
|
ecaresnet101d |
β |
ecaresnet101d_pruned |
β |
ecaresnet200d |
β |
ecaresnet269d |
β |
ecaresnet26t |
β |
ecaresnet50d |
β |
ecaresnet50d_pruned |
β |
ecaresnet50t |
β |
ecaresnetlight |
β |
ecaresnext26t_32x4d |
β |
ecaresnext50t_32x4d |
β |
efficientnet_b0 |
β |
efficientnet_b0_g16_evos |
β |
efficientnet_b0_g8_gn |
β |
efficientnet_b0_gn |
β |
efficientnet_b1 |
β |
efficientnet_b1_pruned |
β |
efficientnet_b2 |
β |
efficientnet_b2_pruned |
β |
efficientnet_b2a |
β |
efficientnet_b3 |
β |
efficientnet_b3_g8_gn |
β |
efficientnet_b3_gn |
β |
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_rw_t |
β |
efficientnetv2_s |
β |
efficientnetv2_xl |
β |
ese_vovnet19b_dw |
|
ese_vovnet19b_slim |
|
ese_vovnet19b_slim_dw |
|
ese_vovnet39b |
|
ese_vovnet39b_evos |
|
ese_vovnet57b |
|
ese_vovnet99b |
|
fbnetc_100 |
β |
fbnetv3_b |
β |
fbnetv3_d |
β |
fbnetv3_g |
β |
gc_efficientnetv2_rw_t |
β |
gcresnet33ts |
β |
gcresnet50t |
β |
gcresnext26ts |
β |
gcresnext50ts |
β |
gernet_l |
β |
gernet_m |
β |
gernet_s |
β |
ghostnet_050 |
|
ghostnet_100 |
|
ghostnet_130 |
|
halo2botnet50ts_256 |
β |
halonet26t |
β |
halonet50ts |
β |
halonet_h1 |
β |
haloregnetz_b |
β |
hardcorenas_a |
β |
hardcorenas_b |
β |
hardcorenas_c |
β |
hardcorenas_d |
β |
hardcorenas_e |
β |
hardcorenas_f |
β |
hrnet_w18 |
|
hrnet_w18_small |
|
hrnet_w18_small_v2 |
|
hrnet_w18_ssld |
|
hrnet_w30 |
|
hrnet_w32 |
|
hrnet_w40 |
|
hrnet_w44 |
|
hrnet_w48 |
|
hrnet_w48_ssld |
|
hrnet_w64 |
|
inception_resnet_v2 |
|
inception_v3 |
|
inception_v4 |
|
lambda_resnet26rpt_256 |
β |
lambda_resnet26t |
β |
lambda_resnet50ts |
β |
lamhalobotnet50ts_256 |
β |
lcnet_035 |
β |
lcnet_050 |
β |
lcnet_075 |
β |
lcnet_100 |
β |
lcnet_150 |
β |
legacy_senet154 |
|
legacy_seresnet101 |
|
legacy_seresnet152 |
|
legacy_seresnet18 |
|
legacy_seresnet34 |
|
legacy_seresnet50 |
|
legacy_seresnext101_32x4d |
|
legacy_seresnext26_32x4d |
|
legacy_seresnext50_32x4d |
|
legacy_xception |
|
maxvit_base_tf_224 |
|
maxvit_base_tf_384 |
|
maxvit_base_tf_512 |
|
maxvit_large_tf_224 |
|
maxvit_large_tf_384 |
|
maxvit_large_tf_512 |
|
maxvit_nano_rw_256 |
|
maxvit_pico_rw_256 |
|
maxvit_rmlp_base_rw_224 |
|
maxvit_rmlp_base_rw_384 |
|
maxvit_rmlp_nano_rw_256 |
|
maxvit_rmlp_pico_rw_256 |
|
maxvit_rmlp_small_rw_224 |
|
maxvit_rmlp_small_rw_256 |
|
maxvit_rmlp_tiny_rw_256 |
|
maxvit_small_tf_224 |
|
maxvit_small_tf_384 |
|
maxvit_small_tf_512 |
|
maxvit_tiny_pm_256 |
|
maxvit_tiny_rw_224 |
|
maxvit_tiny_rw_256 |
|
maxvit_tiny_tf_224 |
|
maxvit_tiny_tf_384 |
|
maxvit_tiny_tf_512 |
|
maxvit_xlarge_tf_224 |
|
maxvit_xlarge_tf_384 |
|
maxvit_xlarge_tf_512 |
|
maxxvit_rmlp_nano_rw_256 |
|
maxxvit_rmlp_small_rw_256 |
|
maxxvit_rmlp_tiny_rw_256 |
|
maxxvitv2_nano_rw_256 |
|
maxxvitv2_rmlp_base_rw_224 |
|
maxxvitv2_rmlp_base_rw_384 |
|
maxxvitv2_rmlp_large_rw_224 |
|
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_035 |
β |
mobilenetv2_050 |
β |
mobilenetv2_075 |
β |
mobilenetv2_100 |
β |
mobilenetv2_110d |
β |
mobilenetv2_120d |
β |
mobilenetv2_140 |
β |
mobilenetv3_large_075 |
β |
mobilenetv3_large_100 |
β |
mobilenetv3_rw |
β |
mobilenetv3_small_050 |
β |
mobilenetv3_small_075 |
β |
mobilenetv3_small_100 |
β |
mobilevit_s |
β |
mobilevit_xs |
β |
mobilevit_xxs |
β |
mobilevitv2_050 |
β |
mobilevitv2_075 |
β |
mobilevitv2_100 |
β |
mobilevitv2_125 |
β |
mobilevitv2_150 |
β |
mobilevitv2_175 |
β |
mobilevitv2_200 |
β |
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_f1 |
β |
nfnet_f2 |
β |
nfnet_f3 |
β |
nfnet_f4 |
β |
nfnet_f5 |
β |
nfnet_f6 |
β |
nfnet_f7 |
β |
nfnet_l0 |
β |
pnasnet5large |
|
regnetv_040 |
β |
regnetv_064 |
β |
regnetx_002 |
β |
regnetx_004 |
β |
regnetx_004_tv |
β |
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_008_tv |
β |
regnety_016 |
β |
regnety_032 |
β |
regnety_040 |
β |
regnety_040_sgn |
β |
regnety_064 |
β |
regnety_080 |
β |
regnety_080_tv |
β |
regnety_120 |
β |
regnety_1280 |
β |
regnety_160 |
β |
regnety_2560 |
β |
regnety_320 |
β |
regnety_640 |
β |
regnetz_005 |
β |
regnetz_040 |
β |
regnetz_040_h |
β |
regnetz_b16 |
β |
regnetz_b16_evos |
β |
regnetz_c16 |
β |
regnetz_c16_evos |
β |
regnetz_d32 |
β |
regnetz_d8 |
β |
regnetz_d8_evos |
β |
regnetz_e8 |
β |
repvgg_a2 |
β |
repvgg_b0 |
β |
repvgg_b1 |
β |
repvgg_b1g4 |
β |
repvgg_b2 |
β |
repvgg_b2g4 |
β |
repvgg_b3 |
β |
repvgg_b3g4 |
β |
res2net101_26w_4s |
β |
res2net101d |
β |
res2net50_14w_8s |
β |
res2net50_26w_4s |
β |
res2net50_26w_6s |
β |
res2net50_26w_8s |
β |
res2net50_48w_2s |
β |
res2net50d |
β |
res2next50 |
β |
resnest101e |
β |
resnest14d |
β |
resnest200e |
β |
resnest269e |
β |
resnest26d |
β |
resnest50d |
β |
resnest50d_1s4x24d |
β |
resnest50d_4s2x40d |
β |
resnet101 |
β |
resnet101c |
β |
resnet101d |
β |
resnet101s |
β |
resnet10t |
β |
resnet14t |
β |
resnet152 |
β |
resnet152c |
β |
resnet152d |
β |
resnet152s |
β |
resnet18 |
β |
resnet18d |
β |
resnet200 |
β |
resnet200d |
β |
resnet26 |
β |
resnet26d |
β |
resnet26t |
β |
resnet32ts |
β |
resnet33ts |
β |
resnet34 |
β |
resnet34d |
β |
resnet50 |
β |
resnet50_gn |
β |
resnet50c |
β |
resnet50d |
β |
resnet50s |
β |
resnet50t |
β |
resnet51q |
β |
resnet61q |
β |
resnetaa101d |
β |
resnetaa34d |
β |
resnetaa50 |
β |
resnetaa50d |
β |
resnetblur101d |
β |
resnetblur18 |
β |
resnetblur50 |
β |
resnetblur50d |
β |
resnetrs101 |
β |
resnetrs152 |
β |
resnetrs200 |
β |
resnetrs270 |
β |
resnetrs350 |
β |
resnetrs420 |
β |
resnetrs50 |
β |
resnetv2_101 |
β |
resnetv2_101d |
β |
resnetv2_101x1_bit |
β |
resnetv2_101x3_bit |
β |
resnetv2_152 |
β |
resnetv2_152d |
β |
resnetv2_152x2_bit |
β |
resnetv2_152x4_bit |
β |
resnetv2_50 |
β |
resnetv2_50d |
β |
resnetv2_50d_evos |
β |
resnetv2_50d_frn |
β |
resnetv2_50d_gn |
β |
resnetv2_50t |
β |
resnetv2_50x1_bit |
β |
resnetv2_50x3_bit |
β |
resnext101_32x16d |
β |
resnext101_32x32d |
β |
resnext101_32x4d |
β |
resnext101_32x8d |
β |
resnext101_64x4d |
β |
resnext26ts |
β |
resnext50_32x4d |
β |
resnext50d_32x4d |
β |
rexnet_100 |
β |
rexnet_130 |
β |
rexnet_150 |
β |
rexnet_200 |
β |
rexnet_300 |
β |
rexnetr_100 |
β |
rexnetr_130 |
β |
rexnetr_150 |
β |
rexnetr_200 |
β |
rexnetr_300 |
β |
sebotnet33ts_256 |
β |
sehalonet33ts |
β |
semnasnet_050 |
β |
semnasnet_075 |
β |
semnasnet_100 |
β |
semnasnet_140 |
β |
senet154 |
β |
seresnet101 |
β |
seresnet152 |
β |
seresnet152d |
β |
seresnet18 |
β |
seresnet200d |
β |
seresnet269d |
β |
seresnet33ts |
β |
seresnet34 |
β |
seresnet50 |
β |
seresnet50t |
β |
seresnetaa50d |
β |
seresnext101_32x4d |
β |
seresnext101_32x8d |
β |
seresnext101_64x4d |
β |
seresnext101d_32x8d |
β |
seresnext26d_32x4d |
β |
seresnext26t_32x4d |
β |
seresnext26tn_32x4d |
β |
seresnext26ts |
β |
seresnext50_32x4d |
β |
seresnextaa101d_32x8d |
β |
skresnet18 |
β |
skresnet34 |
β |
skresnet50 |
β |
skresnet50d |
β |
skresnext50_32x4d |
β |
spnasnet_100 |
β |
tf_efficientnet_b0 |
β |
tf_efficientnet_b1 |
β |
tf_efficientnet_b2 |
β |
tf_efficientnet_b3 |
β |
tf_efficientnet_b4 |
β |
tf_efficientnet_b5 |
β |
tf_efficientnet_b6 |
β |
tf_efficientnet_b7 |
β |
tf_efficientnet_b8 |
β |
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 |
β |
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_m |
β |
tf_efficientnetv2_s |
β |
tf_efficientnetv2_xl |
β |
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 |
β |
tinynet_a |
β |
tinynet_b |
β |
tinynet_c |
β |
tinynet_d |
β |
tinynet_e |
β |
vovnet39a |
|
vovnet57a |
|
wide_resnet101_2 |
β |
wide_resnet50_2 |
β |
xception41 |
β |
xception41p |
β |
xception65 |
β |
xception65p |
β |
xception71 |
β |