Awesome

Istanbul Dataset

This repo contains weights of Unet++ model with SE-ResNeXt101 encoder trained with Istanbul, Inria and Massachusetts datasets seperately. Trainings have been realized using PyTorch and segmentation models library (https://github.com/qubvel/segmentation_models.pytorch) We also provide an inference notebook to run prediction on GeoTiff images. This notebook also outputs prediction images as GeoTiff.

Update:

We have addeed more weights of different architectures trained with Istanbul dataset.

You can use the following links to download weights files:

• Unet++ trained with Istanbul Dataset: Download
• Unet++ trained with Inria Dataset: Download
• Unet++ trained with Massachusetts Dataset: Download

New Weights trained with Istanbul Dataset:

• Unet++ with InceptionResNetv2 encoder: Download
• Unet++ with EfficientNet-b6 encoder: Download
• UNet with SE-ResNeXt101 encoder: Download
• UNet with InceptionResNetv2 encoder: Download
• UNet with EfficientNet-b6 encoder: Download
• DeepLabv3+ with SE-ResNeXt101 encoder: Download

To run the notebook, following libraries are required:

• torch == 1.7.1
• segmentation-models-pytorch == 0.1.3
• scikit-image == 0.18.1
• GDAL == 3.2.1
• tifffile == 2021.2.1

Citation

Please cite the following study if you make use of our weights: Paper