Convolutional Neural Network for Road Segmentation

Abstract

We trained a convolutional neural network to perform image segmentation on aerial images of urban areas. The objective of the model is to be able to identify and separate roads from the surrounding environment. The network comprises four convolutional layers of increasing depth followed by a fully connected one. Dropout and image augmentation have been used to improve the performance of the model as well as diminish overfitting.

This project was realized as part of the CS-433 Machine Learning course at EPFL.

Links

Read the report

Browse the source code

 

Input fed to the model
Output of the model. Areas not detected as roads have been coloured in red