| SpaceNet7 Buildings Detection |
Content
This project studied building footprints detection in satellite imagery, a baseline need for many organizations that can enable many types of analyses. It can be used as a prxoy for population statistics, informe disasters and crises (i.e, disaster and response planning and preparedness: what infrastructure might be at risk), understanding buildings change over time (i.e,rate of urbanization, changes in green spaces). From a humanitarian perspective, such information has a significant impact on preparation, mitigation, and triaging responses and resource allocations to accelerate recovery.
The goal of this Notebook is to develop a neural network based solution to detect buildings in the images provided to accelarate mapping. The analysis is based on a curated subset of the SpaceNet7 dataset. This dataset is hosted on AWS as a Public Dataset.
This dataset encapsulates the moderate resolution (4m/pixel) EO data, each with 4 channel data (red, green, blue and near-infrared) and a corresponding ground truth mask. We explore the above problem by extracting semantic maps of buildings in medium resolution EO images. Devising it as a supervised learning problem, the state of the art deep neural network architecture, namely, UNet, that is commonly used for the the task of semantic segmentation was designed, implemented and experimentally evaluated. Because the initial dataset is very small (56 images) we have used patch learning. We basically generated a larger dataset by extracting random crops from the original images.
The main tasks performed are:
- Preprocess/transform data to the format that can be consumed by the model;
- Implement UNet to detect building footprints;
- Train and evaluate the model's results
The experiments carried out in this project are built on top of pytorch, an open source machine learning framework in python, and MONAI framework - a PyTorch-based open-source foundation for deep learning in healthcare imaging.The resources provided on the kaggle platform were used to run these experiments. Kaggle provides free access to NVIDIA TESLA P100 GPUs. These can be used with a quota limit of 30 hours per week. These GPUs are useful for training deep learning models, and they take advantage of GPU-accelerated libraries (e.g. TensorFlow, PyTorch, etc)
UNet consists of an 'encoding' and a 'decoding' part. The encoder is an alternating series of convolution-pooling layers, that extract features from the input, very much like an ordinary classifier. The decoder produces a segmentation map, based on the features derived in the encoder, by alternating transposed convolution layers (or upsampling) and convolution layers. UNet introduces skip-connections between encoder and decoder, at levels where the feature maps have the same lateral extent (number of channels). This enables the decoder to access information from the encoder, such as the general features (edges...) in the original images. The UNet network depicted in this paper is the one we used in our project. The source code for this network implemented using MONAI is provided here . I have also implemented UNet from scratch using plain pytorch (provide below). The MONAI implementation outperformed the the later. Therefore, I decied to use the MONAI UNet. The U-Net that we are using comprises 5 levels. At each stage two convolution operations are applied, each followed by an Instance normalization and the leaky ReLU activation.
We are using the U-Net model because: * It is a very simple architecture, which means it is easy to implement and to debug. * Compared to other architectures, its simplicity makes it faster (less trainable parameters). This is advantageous, as we want to apply the model to a relatively large dataset within a reasonable amount of time to get a first intuition about the data. * We can use as much input channels as we want and we are not restricted to only the 3 channels of the standard RGB images. We can also use different sizes of images as in our case (train on (C, 480, 480) image patches and evaluate on the full (C, 1024, 1024)images); * It has been shown in various cases: UNet , DeepUnet , Deep Residual U-Net , that the U-Net is a very capable network with high performance even for satellite imagery.
In this project you will find:
- requirements.txt it contains all the necessary libraries;
- scripts contains a modular code;
- trained_models contains the best model based on the dice score and the last trained model;
- output_samples contains some predictions and the corresponding ground truth images.
This section provides instructions for the model development phase.
Download SpaceNet7 data
The dataset (satellite images and their corresponding masks) used in this project is based on the SpaceNet7 dataset. The original data is hosted at:
s3://spacenet-dataset/spacenet/SN7_buildings/
Prepare environment
# install MONAI
pip install monai # import the necessary libraries
import torch
import matplotlib.pyplot as plt
import numpy as np# git clone source
!git clone https://github.com/Rim-chan/SpaceNet7-Buildings-Detection.gitTrain segmentation model
!python ./SpaceNet7-Buildings-Detection/scripts/main.py --base_dir "../input/SpaceNet7_curated_dataset" --num_epochs 10 --exec_mode 'train'Test segmentation model
!python ./SpaceNet7-Buildings-Detection/scripts/main.py --base_dir "../input/SpaceNet7_curated_dataset" --exec_mode 'evaluate' --ckpt_path './last.ckpt'Load and display some samples
preds = np.load('./predictions.npy') #(6, 1, 1024, 1024)
lbls = np.load('./labels.npy') #(6, 1, 1024, 1024)
# plot some examples
fig, ax = plt.subplots(1,2, figsize = (20,10))
ax[0].imshow(preds[3][0], cmap='gray')
ax[1].imshow(lbls[3][0], cmap='gray') 
