segmentation_models_pytorch/docs/encoders.rst

🏔 Available Encoders
=====================

ResNet
~~~~~~

+-------------+-------------------------+-------------+
| Encoder     | Weights                 | Params, M   |
+=============+=========================+=============+
| resnet18    | imagenet / ssl / swsl   | 11M         |
+-------------+-------------------------+-------------+
| resnet34    | imagenet                | 21M         |
+-------------+-------------------------+-------------+
| resnet50    | imagenet / ssl / swsl   | 23M         |
+-------------+-------------------------+-------------+
| resnet101   | imagenet                | 42M         |
+-------------+-------------------------+-------------+
| resnet152   | imagenet                | 58M         |
+-------------+-------------------------+-------------+

ResNeXt
~~~~~~~

+----------------------+-------------------------------------+-------------+
| Encoder              | Weights                             | Params, M   |
+======================+=====================================+=============+
| resnext50\_32x4d     | imagenet / ssl / swsl               | 22M         |
+----------------------+-------------------------------------+-------------+
| resnext101\_32x4d    | ssl / swsl                          | 42M         |
+----------------------+-------------------------------------+-------------+
| resnext101\_32x8d    | imagenet / instagram / ssl / swsl   | 86M         |
+----------------------+-------------------------------------+-------------+
| resnext101\_32x16d   | instagram / ssl / swsl              | 191M        |
+----------------------+-------------------------------------+-------------+
| resnext101\_32x32d   | instagram                           | 466M        |
+----------------------+-------------------------------------+-------------+
| resnext101\_32x48d   | instagram                           | 826M        |
+----------------------+-------------------------------------+-------------+

ResNeSt
~~~~~~~

+----------------------------+------------+-------------+
| Encoder                    | Weights    | Params, M   |
+============================+============+=============+
| timm-resnest14d            | imagenet   | 8M          |
+----------------------------+------------+-------------+
| timm-resnest26d            | imagenet   | 15M         |
+----------------------------+------------+-------------+
| timm-resnest50d            | imagenet   | 25M         |
+----------------------------+------------+-------------+
| timm-resnest101e           | imagenet   | 46M         |
+----------------------------+------------+-------------+
| timm-resnest200e           | imagenet   | 68M         |
+----------------------------+------------+-------------+
| timm-resnest269e           | imagenet   | 108M        |
+----------------------------+------------+-------------+
| timm-resnest50d\_4s2x40d   | imagenet   | 28M         |
+----------------------------+------------+-------------+
| timm-resnest50d\_1s4x24d   | imagenet   | 23M         |
+----------------------------+------------+-------------+

Res2Ne(X)t
~~~~~~~~~~

+----------------------------+------------+-------------+
| Encoder                    | Weights    | Params, M   |
+============================+============+=============+
| timm-res2net50\_26w\_4s    | imagenet   | 23M         |
+----------------------------+------------+-------------+
| timm-res2net101\_26w\_4s   | imagenet   | 43M         |
+----------------------------+------------+-------------+
| timm-res2net50\_26w\_6s    | imagenet   | 35M         |
+----------------------------+------------+-------------+
| timm-res2net50\_26w\_8s    | imagenet   | 46M         |
+----------------------------+------------+-------------+
| timm-res2net50\_48w\_2s    | imagenet   | 23M         |
+----------------------------+------------+-------------+
| timm-res2net50\_14w\_8s    | imagenet   | 23M         |
+----------------------------+------------+-------------+
| timm-res2next50            | imagenet   | 22M         |
+----------------------------+------------+-------------+

RegNet(x/y)
~~~~~~~~~~~

+---------------------+------------+-------------+
| Encoder             | Weights    | Params, M   |
+=====================+============+=============+
| timm-regnetx\_002   | imagenet   | 2M          |
+---------------------+------------+-------------+
| timm-regnetx\_004   | imagenet   | 4M          |
+---------------------+------------+-------------+
| timm-regnetx\_006   | imagenet   | 5M          |
+---------------------+------------+-------------+
| timm-regnetx\_008   | imagenet   | 6M          |
+---------------------+------------+-------------+
| timm-regnetx\_016   | imagenet   | 8M          |
+---------------------+------------+-------------+
| timm-regnetx\_032   | imagenet   | 14M         |
+---------------------+------------+-------------+
| timm-regnetx\_040   | imagenet   | 20M         |
+---------------------+------------+-------------+
| timm-regnetx\_064   | imagenet   | 24M         |
+---------------------+------------+-------------+
| timm-regnetx\_080   | imagenet   | 37M         |
+---------------------+------------+-------------+
| timm-regnetx\_120   | imagenet   | 43M         |
+---------------------+------------+-------------+
| timm-regnetx\_160   | imagenet   | 52M         |
+---------------------+------------+-------------+
| timm-regnetx\_320   | imagenet   | 105M        |
+---------------------+------------+-------------+
| timm-regnety\_002   | imagenet   | 2M          |
+---------------------+------------+-------------+
| timm-regnety\_004   | imagenet   | 3M          |
+---------------------+------------+-------------+
| timm-regnety\_006   | imagenet   | 5M          |
+---------------------+------------+-------------+
| timm-regnety\_008   | imagenet   | 5M          |
+---------------------+------------+-------------+
| timm-regnety\_016   | imagenet   | 10M         |
+---------------------+------------+-------------+
| timm-regnety\_032   | imagenet   | 17M         |
+---------------------+------------+-------------+
| timm-regnety\_040   | imagenet   | 19M         |
+---------------------+------------+-------------+
| timm-regnety\_064   | imagenet   | 29M         |
+---------------------+------------+-------------+
| timm-regnety\_080   | imagenet   | 37M         |
+---------------------+------------+-------------+
| timm-regnety\_120   | imagenet   | 49M         |
+---------------------+------------+-------------+
| timm-regnety\_160   | imagenet   | 80M         |
+---------------------+------------+-------------+
| timm-regnety\_320   | imagenet   | 141M        |
+---------------------+------------+-------------+

SE-Net
~~~~~~

+-------------------------+------------+-------------+
| Encoder                 | Weights    | Params, M   |
+=========================+============+=============+
| senet154                | imagenet   | 113M        |
+-------------------------+------------+-------------+
| se\_resnet50            | imagenet   | 26M         |
+-------------------------+------------+-------------+
| se\_resnet101           | imagenet   | 47M         |
+-------------------------+------------+-------------+
| se\_resnet152           | imagenet   | 64M         |
+-------------------------+------------+-------------+
| se\_resnext50\_32x4d    | imagenet   | 25M         |
+-------------------------+------------+-------------+
| se\_resnext101\_32x4d   | imagenet   | 46M         |
+-------------------------+------------+-------------+

SK-ResNe(X)t
~~~~~~~~~~~~

+---------------------------+------------+-------------+
| Encoder                   | Weights    | Params, M   |
+===========================+============+=============+
| timm-skresnet18           | imagenet   | 11M         |
+---------------------------+------------+-------------+
| timm-skresnet34           | imagenet   | 21M         |
+---------------------------+------------+-------------+
| timm-skresnext50\_32x4d   | imagenet   | 25M         |
+---------------------------+------------+-------------+

DenseNet
~~~~~~~~

+---------------+------------+-------------+
| Encoder       | Weights    | Params, M   |
+===============+============+=============+
| densenet121   | imagenet   | 6M          |
+---------------+------------+-------------+
| densenet169   | imagenet   | 12M         |
+---------------+------------+-------------+
| densenet201   | imagenet   | 18M         |
+---------------+------------+-------------+
| densenet161   | imagenet   | 26M         |
+---------------+------------+-------------+

Inception
~~~~~~~~~

+---------------------+----------------------------------+-------------+
| Encoder             | Weights                          | Params, M   |
+=====================+==================================+=============+
| inceptionresnetv2   | imagenet / imagenet+background   | 54M         |
+---------------------+----------------------------------+-------------+
| inceptionv4         | imagenet / imagenet+background   | 41M         |
+---------------------+----------------------------------+-------------+
| xception            | imagenet                         | 22M         |
+---------------------+----------------------------------+-------------+

EfficientNet
~~~~~~~~~~~~

+------------------------+--------------------------------------+-------------+
| Encoder                | Weights                              | Params, M   |
+========================+======================================+=============+
| efficientnet-b0        | imagenet                             | 4M          |
+------------------------+--------------------------------------+-------------+
| efficientnet-b1        | imagenet                             | 6M          |
+------------------------+--------------------------------------+-------------+
| efficientnet-b2        | imagenet                             | 7M          |
+------------------------+--------------------------------------+-------------+
| efficientnet-b3        | imagenet                             | 10M         |
+------------------------+--------------------------------------+-------------+
| efficientnet-b4        | imagenet                             | 17M         |
+------------------------+--------------------------------------+-------------+
| efficientnet-b5        | imagenet                             | 28M         |
+------------------------+--------------------------------------+-------------+
| efficientnet-b6        | imagenet                             | 40M         |
+------------------------+--------------------------------------+-------------+
| efficientnet-b7        | imagenet                             | 63M         |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b0   | imagenet / advprop / noisy-student   | 4M          |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b1   | imagenet / advprop / noisy-student   | 6M          |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b2   | imagenet / advprop / noisy-student   | 7M          |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b3   | imagenet / advprop / noisy-student   | 10M         |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b4   | imagenet / advprop / noisy-student   | 17M         |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b5   | imagenet / advprop / noisy-student   | 28M         |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b6   | imagenet / advprop / noisy-student   | 40M         |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b7   | imagenet / advprop / noisy-student   | 63M         |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-b8   | imagenet / advprop                   | 84M         |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-l2   | noisy-student                        | 474M        |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-lite0| imagenet                             | 4M          |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-lite1| imagenet                             | 4M          |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-lite2| imagenet                             | 6M          |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-lite3| imagenet                             | 8M          |
+------------------------+--------------------------------------+-------------+
| timm-efficientnet-lite4| imagenet                             | 13M         |
+------------------------+--------------------------------------+-------------+

MobileNet
~~~~~~~~~

+-----------------+------------+-------------+
| Encoder         | Weights    | Params, M   |
+=================+============+=============+
| mobilenet\_v2   | imagenet   | 2M          |
+-----------------+------------+-------------+

DPN
~~~

+-----------+---------------+-------------+
| Encoder   | Weights       | Params, M   |
+===========+===============+=============+
| dpn68     | imagenet      | 11M         |
+-----------+---------------+-------------+
| dpn68b    | imagenet+5k   | 11M         |
+-----------+---------------+-------------+
| dpn92     | imagenet+5k   | 34M         |
+-----------+---------------+-------------+
| dpn98     | imagenet      | 58M         |
+-----------+---------------+-------------+
| dpn107    | imagenet+5k   | 84M         |
+-----------+---------------+-------------+
| dpn131    | imagenet      | 76M         |
+-----------+---------------+-------------+

VGG
~~~

+-------------+------------+-------------+
| Encoder     | Weights    | Params, M   |
+=============+============+=============+
| vgg11       | imagenet   | 9M          |
+-------------+------------+-------------+
| vgg11\_bn   | imagenet   | 9M          |
+-------------+------------+-------------+
| vgg13       | imagenet   | 9M          |
+-------------+------------+-------------+
| vgg13\_bn   | imagenet   | 9M          |
+-------------+------------+-------------+
| vgg16       | imagenet   | 14M         |
+-------------+------------+-------------+
| vgg16\_bn   | imagenet   | 14M         |
+-------------+------------+-------------+
| vgg19       | imagenet   | 20M         |
+-------------+------------+-------------+
| vgg19\_bn   | imagenet   | 20M         |
+-------------+------------+-------------+