gitextract_j1x_q930/ ├── .devcontainer/ │ └── devcontainer.json ├── .github/ │ ├── FUNDING.yml │ ├── codecov.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 │ ├── 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 ├── 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