Skip to content
Using Pytorch implementation of U-Net architecture for road and building extraction
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
etc add image for tensorboard Mar 6, 2018
notebooks
src
.gitignore added logging and checkpointing Feb 15, 2018
LICENSE
README.md

README.md

Road Extraction Using PyTorch

alt text

This repository contains the code used for this project. Feel free to click on the link to read more about the details of the project!

The main dataset used to train the network was the Road and Building Detection Dataset created by Volodymyr Mnih as part of his PhD thesis at the University of Toronto. Below is an example of a training image overlayed with the road network mask.

alt text

The main architecture is the U-Net architecture that was used initially for biomedical image segmentation, but has since been successful in other tasks such as satellite imagery segmentation.

alt text

During the training process, I mainly used Tensorboard to keep track of the different runs.

alt text

Below is an example of the output from the network (the left image is the input RGB satellite image, the middle is the ground truth label, and the right side shows the model prediction).

alt text

You can’t perform that action at this time.