Road Extraction Using PyTorch
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.
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.
During the training process, I mainly used Tensorboard to keep track of the different runs.
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).