├── .devcontainer └── devcontainer.json ├── .github ├── FUNDING.yml ├── dependabot.yml └── workflows │ ├── pypi.yml │ ├── stale.yaml │ └── tests.yml ├── .gitignore ├── .readthedocs.yaml ├── HALLOFFAME.md ├── LICENSE ├── Makefile ├── README.md ├── docs ├── Makefile ├── conf.py ├── encoders.rst ├── encoders_dpt.rst ├── encoders_timm.rst ├── index.rst ├── insights.rst ├── install.rst ├── logo.png ├── losses.rst ├── make.bat ├── metrics.rst ├── models.rst ├── quickstart.rst └── save_load.rst ├── examples ├── binary_segmentation_buildings.py ├── binary_segmentation_intro.ipynb ├── camvid_segmentation_multiclass.ipynb ├── cars segmentation (camvid).ipynb ├── convert_to_onnx.ipynb ├── dpt_inference_pretrained.ipynb ├── save_load_model_and_share_with_hf_hub.ipynb ├── segformer_inference_pretrained.ipynb └── upernet_inference_pretrained.ipynb ├── licenses ├── LICENSES.md ├── LICENSE_apache.md ├── LICENSE_apple.md └── LICENSE_nvidia.md ├── misc ├── generate_table.py ├── generate_table_timm.py └── generate_test_models.py ├── pics ├── logo-small-h300.png └── logo-small-w300.png ├── pyproject.toml ├── requirements ├── docs.txt ├── minimum.old ├── required.txt └── test.txt ├── scripts └── models-conversions │ ├── dpt-original-to-smp.py │ ├── segformer-original-decoder-to-smp.py │ └── upernet-hf-to-smp.py ├── segmentation_models_pytorch ├── __init__.py ├── __version__.py ├── base │ ├── __init__.py │ ├── heads.py │ ├── hub_mixin.py │ ├── initialization.py │ ├── model.py │ ├── modules.py │ └── utils.py ├── datasets │ ├── __init__.py │ └── oxford_pet.py ├── decoders │ ├── __init__.py │ ├── deeplabv3 │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── dpt │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── fpn │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── linknet │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── manet │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── pan │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── pspnet │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── segformer │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── unet │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ ├── unetplusplus │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py │ └── upernet │ │ ├── __init__.py │ │ ├── decoder.py │ │ └── model.py ├── encoders │ ├── __init__.py │ ├── _base.py │ ├── _dpn.py │ ├── _efficientnet.py │ ├── _inceptionresnetv2.py │ ├── _inceptionv4.py │ ├── _legacy_pretrained_settings.py │ ├── _preprocessing.py │ ├── _senet.py │ ├── _utils.py │ ├── _xception.py │ ├── densenet.py │ ├── dpn.py │ ├── efficientnet.py │ ├── inceptionresnetv2.py │ ├── inceptionv4.py │ ├── mix_transformer.py │ ├── mobilenet.py │ ├── mobileone.py │ ├── resnet.py │ ├── senet.py │ ├── timm_efficientnet.py │ ├── timm_sknet.py │ ├── timm_universal.py │ ├── timm_vit.py │ ├── vgg.py │ └── xception.py ├── losses │ ├── __init__.py │ ├── _functional.py │ ├── constants.py │ ├── dice.py │ ├── focal.py │ ├── jaccard.py │ ├── lovasz.py │ ├── mcc.py │ ├── soft_bce.py │ ├── soft_ce.py │ └── tversky.py ├── metrics │ ├── __init__.py │ └── functional.py └── utils │ ├── __init__.py │ ├── base.py │ ├── functional.py │ ├── losses.py │ ├── meter.py │ ├── metrics.py │ └── train.py └── tests ├── __init__.py ├── base ├── test_freeze_encoder.py └── test_modules.py ├── conftest.py ├── encoders ├── __init__.py ├── base.py ├── test_batchnorm_deprecation.py ├── test_common.py ├── test_pretrainedmodels_encoders.py ├── test_smp_encoders.py ├── test_timm_ported_encoders.py ├── test_timm_universal.py ├── test_timm_vit_encoders.py └── test_torchvision_encoders.py ├── models ├── __init__.py ├── base.py ├── test_deeplab.py ├── test_dpt.py ├── test_fpn.py ├── test_linknet.py ├── test_manet.py ├── test_pan.py ├── test_psp.py ├── test_segformer.py ├── test_unet.py ├── test_unetplusplus.py └── test_upernet.py ├── test_base.py ├── test_losses.py ├── test_preprocessing.py └── utils.py /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/.github/workflows/pypi.yml -------------------------------------------------------------------------------- /.github/workflows/stale.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/.github/workflows/stale.yaml -------------------------------------------------------------------------------- /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /HALLOFFAME.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/HALLOFFAME.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/README.md -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/encoders.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/encoders.rst -------------------------------------------------------------------------------- /docs/encoders_dpt.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/encoders_dpt.rst -------------------------------------------------------------------------------- /docs/encoders_timm.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/encoders_timm.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/insights.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/insights.rst -------------------------------------------------------------------------------- /docs/install.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/install.rst -------------------------------------------------------------------------------- /docs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/logo.png -------------------------------------------------------------------------------- /docs/losses.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/losses.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/metrics.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/metrics.rst -------------------------------------------------------------------------------- /docs/models.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/models.rst -------------------------------------------------------------------------------- /docs/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/quickstart.rst -------------------------------------------------------------------------------- /docs/save_load.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/docs/save_load.rst -------------------------------------------------------------------------------- /examples/binary_segmentation_buildings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/binary_segmentation_buildings.py -------------------------------------------------------------------------------- /examples/binary_segmentation_intro.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/binary_segmentation_intro.ipynb -------------------------------------------------------------------------------- /examples/camvid_segmentation_multiclass.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/camvid_segmentation_multiclass.ipynb -------------------------------------------------------------------------------- /examples/cars segmentation (camvid).ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/cars segmentation (camvid).ipynb -------------------------------------------------------------------------------- /examples/convert_to_onnx.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/convert_to_onnx.ipynb -------------------------------------------------------------------------------- /examples/dpt_inference_pretrained.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/dpt_inference_pretrained.ipynb -------------------------------------------------------------------------------- /examples/save_load_model_and_share_with_hf_hub.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/save_load_model_and_share_with_hf_hub.ipynb -------------------------------------------------------------------------------- /examples/segformer_inference_pretrained.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/segformer_inference_pretrained.ipynb -------------------------------------------------------------------------------- /examples/upernet_inference_pretrained.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/examples/upernet_inference_pretrained.ipynb -------------------------------------------------------------------------------- /licenses/LICENSES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/licenses/LICENSES.md -------------------------------------------------------------------------------- /licenses/LICENSE_apache.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/licenses/LICENSE_apache.md -------------------------------------------------------------------------------- /licenses/LICENSE_apple.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/licenses/LICENSE_apple.md -------------------------------------------------------------------------------- /licenses/LICENSE_nvidia.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/licenses/LICENSE_nvidia.md -------------------------------------------------------------------------------- /misc/generate_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/misc/generate_table.py -------------------------------------------------------------------------------- /misc/generate_table_timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/misc/generate_table_timm.py -------------------------------------------------------------------------------- /misc/generate_test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/misc/generate_test_models.py -------------------------------------------------------------------------------- /pics/logo-small-h300.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/pics/logo-small-h300.png -------------------------------------------------------------------------------- /pics/logo-small-w300.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/pics/logo-small-w300.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/requirements/docs.txt -------------------------------------------------------------------------------- /requirements/minimum.old: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/requirements/minimum.old -------------------------------------------------------------------------------- /requirements/required.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/requirements/required.txt -------------------------------------------------------------------------------- /requirements/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/requirements/test.txt -------------------------------------------------------------------------------- /scripts/models-conversions/dpt-original-to-smp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/scripts/models-conversions/dpt-original-to-smp.py -------------------------------------------------------------------------------- /scripts/models-conversions/segformer-original-decoder-to-smp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/scripts/models-conversions/segformer-original-decoder-to-smp.py -------------------------------------------------------------------------------- /scripts/models-conversions/upernet-hf-to-smp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/scripts/models-conversions/upernet-hf-to-smp.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/__version__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.5.1.dev0" 2 | -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/base/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/base/heads.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/hub_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/base/hub_mixin.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/initialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/base/initialization.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/base/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/base/modules.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/base/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/base/utils.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/datasets/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/datasets/oxford_pet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/datasets/oxford_pet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/deeplabv3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/deeplabv3/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/deeplabv3/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/deeplabv3/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/deeplabv3/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/deeplabv3/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/dpt/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import DPT 2 | 3 | __all__ = ["DPT"] 4 | -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/dpt/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/dpt/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/dpt/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/dpt/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/fpn/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import FPN 2 | 3 | __all__ = ["FPN"] 4 | -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/fpn/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/fpn/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/fpn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/fpn/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/linknet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/linknet/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/linknet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/linknet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/linknet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/linknet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/manet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/manet/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/manet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/manet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/manet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/manet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/pan/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import PAN 2 | 3 | __all__ = ["PAN"] 4 | -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/pan/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/pan/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/pan/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/pan/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/pspnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/pspnet/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/pspnet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/pspnet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/pspnet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/pspnet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/segformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/segformer/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/segformer/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/segformer/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/segformer/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/segformer/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/unet/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import Unet 2 | 3 | __all__ = ["Unet"] 4 | -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/unet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/unet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/unet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/unet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/unetplusplus/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/unetplusplus/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/unetplusplus/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/unetplusplus/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/unetplusplus/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/unetplusplus/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/upernet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/upernet/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/upernet/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/upernet/decoder.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/decoders/upernet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/decoders/upernet/model.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_base.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_dpn.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_efficientnet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_inceptionresnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_inceptionresnetv2.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_inceptionv4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_inceptionv4.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_legacy_pretrained_settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_legacy_pretrained_settings.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_preprocessing.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_senet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_utils.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/_xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/_xception.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/densenet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/dpn.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/efficientnet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/inceptionresnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/inceptionresnetv2.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/inceptionv4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/inceptionv4.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/mix_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/mix_transformer.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/mobilenet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/mobileone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/mobileone.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/resnet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/senet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/timm_efficientnet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_sknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/timm_sknet.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_universal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/timm_universal.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/timm_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/timm_vit.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/vgg.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/encoders/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/encoders/xception.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/_functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/_functional.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/constants.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/dice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/dice.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/focal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/focal.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/jaccard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/jaccard.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/lovasz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/lovasz.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/mcc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/mcc.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/soft_bce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/soft_bce.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/soft_ce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/soft_ce.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/losses/tversky.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/losses/tversky.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/metrics/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/metrics/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/metrics/functional.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/utils/__init__.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/utils/base.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/utils/functional.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/utils/losses.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/utils/meter.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/utils/metrics.py -------------------------------------------------------------------------------- /segmentation_models_pytorch/utils/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/segmentation_models_pytorch/utils/train.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/base/test_freeze_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/base/test_freeze_encoder.py -------------------------------------------------------------------------------- /tests/base/test_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/base/test_modules.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/encoders/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/base.py -------------------------------------------------------------------------------- /tests/encoders/test_batchnorm_deprecation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/test_batchnorm_deprecation.py -------------------------------------------------------------------------------- /tests/encoders/test_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/test_common.py -------------------------------------------------------------------------------- /tests/encoders/test_pretrainedmodels_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/test_pretrainedmodels_encoders.py -------------------------------------------------------------------------------- /tests/encoders/test_smp_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/test_smp_encoders.py -------------------------------------------------------------------------------- /tests/encoders/test_timm_ported_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/test_timm_ported_encoders.py -------------------------------------------------------------------------------- /tests/encoders/test_timm_universal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/test_timm_universal.py -------------------------------------------------------------------------------- /tests/encoders/test_timm_vit_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/test_timm_vit_encoders.py -------------------------------------------------------------------------------- /tests/encoders/test_torchvision_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/encoders/test_torchvision_encoders.py -------------------------------------------------------------------------------- /tests/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/base.py -------------------------------------------------------------------------------- /tests/models/test_deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_deeplab.py -------------------------------------------------------------------------------- /tests/models/test_dpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_dpt.py -------------------------------------------------------------------------------- /tests/models/test_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_fpn.py -------------------------------------------------------------------------------- /tests/models/test_linknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_linknet.py -------------------------------------------------------------------------------- /tests/models/test_manet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_manet.py -------------------------------------------------------------------------------- /tests/models/test_pan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_pan.py -------------------------------------------------------------------------------- /tests/models/test_psp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_psp.py -------------------------------------------------------------------------------- /tests/models/test_segformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_segformer.py -------------------------------------------------------------------------------- /tests/models/test_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_unet.py -------------------------------------------------------------------------------- /tests/models/test_unetplusplus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_unetplusplus.py -------------------------------------------------------------------------------- /tests/models/test_upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/models/test_upernet.py -------------------------------------------------------------------------------- /tests/test_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/test_base.py -------------------------------------------------------------------------------- /tests/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/test_losses.py -------------------------------------------------------------------------------- /tests/test_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/test_preprocessing.py -------------------------------------------------------------------------------- /tests/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel-org/segmentation_models.pytorch/HEAD/tests/utils.py --------------------------------------------------------------------------------