├── .github └── FUNDING.yml ├── .gitignore ├── .gitmodules ├── .travis.yml ├── CHANGELOG.md ├── LICENSE ├── MANIFEST.in ├── README.rst ├── __init__.py ├── docs ├── Makefile ├── api.rst ├── conf.py ├── index.rst ├── install.rst ├── support.rst └── tutorial.rst ├── examples ├── binary segmentation (camvid).ipynb └── multiclass segmentation (camvid).ipynb ├── images ├── fpn.png ├── linknet.png ├── logo.png ├── pspnet.png └── unet.png ├── requirements.txt ├── segmentation_models ├── __init__.py ├── __version__.py ├── backbones │ ├── __init__.py │ ├── backbones_factory.py │ ├── inception_resnet_v2.py │ └── inception_v3.py ├── base │ ├── __init__.py │ ├── functional.py │ └── objects.py ├── losses.py ├── metrics.py ├── models │ ├── __init__.py │ ├── _common_blocks.py │ ├── _utils.py │ ├── fpn.py │ ├── linknet.py │ ├── pspnet.py │ └── unet.py └── utils.py ├── setup.py └── tests ├── test_metrics.py ├── test_models.py └── test_utils.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/.travis.yml -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | include README.md LICENSE requirements.txt 2 | -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/README.rst -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | from .segmentation_models import * 2 | -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/docs/api.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/install.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/docs/install.rst -------------------------------------------------------------------------------- /docs/support.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/docs/support.rst -------------------------------------------------------------------------------- /docs/tutorial.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/docs/tutorial.rst -------------------------------------------------------------------------------- /examples/binary segmentation (camvid).ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/examples/binary segmentation (camvid).ipynb -------------------------------------------------------------------------------- /examples/multiclass segmentation (camvid).ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/examples/multiclass segmentation (camvid).ipynb -------------------------------------------------------------------------------- /images/fpn.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/images/fpn.png -------------------------------------------------------------------------------- /images/linknet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/images/linknet.png -------------------------------------------------------------------------------- /images/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/images/logo.png -------------------------------------------------------------------------------- /images/pspnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/images/pspnet.png -------------------------------------------------------------------------------- /images/unet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/images/unet.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/requirements.txt -------------------------------------------------------------------------------- /segmentation_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/__init__.py -------------------------------------------------------------------------------- /segmentation_models/__version__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/__version__.py -------------------------------------------------------------------------------- /segmentation_models/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentation_models/backbones/backbones_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/backbones/backbones_factory.py -------------------------------------------------------------------------------- /segmentation_models/backbones/inception_resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/backbones/inception_resnet_v2.py -------------------------------------------------------------------------------- /segmentation_models/backbones/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/backbones/inception_v3.py -------------------------------------------------------------------------------- /segmentation_models/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/base/__init__.py -------------------------------------------------------------------------------- /segmentation_models/base/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/base/functional.py -------------------------------------------------------------------------------- /segmentation_models/base/objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/base/objects.py -------------------------------------------------------------------------------- /segmentation_models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/losses.py -------------------------------------------------------------------------------- /segmentation_models/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/metrics.py -------------------------------------------------------------------------------- /segmentation_models/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentation_models/models/_common_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/models/_common_blocks.py -------------------------------------------------------------------------------- /segmentation_models/models/_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/models/_utils.py -------------------------------------------------------------------------------- /segmentation_models/models/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/models/fpn.py -------------------------------------------------------------------------------- /segmentation_models/models/linknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/models/linknet.py -------------------------------------------------------------------------------- /segmentation_models/models/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/models/pspnet.py -------------------------------------------------------------------------------- /segmentation_models/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/models/unet.py -------------------------------------------------------------------------------- /segmentation_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/segmentation_models/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/tests/test_metrics.py -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qubvel/segmentation_models/HEAD/tests/test_utils.py --------------------------------------------------------------------------------