RSandAI/Istanbul-Building-Dataset-Benchmark-Building-Extraction-Dataset-and-DL-Models

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.

Jupyter Notebook