Skip to content

Using Pytorch implementation of U-Net architecture for road and building extraction

License

Notifications You must be signed in to change notification settings

jeffwen/road_building_extraction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

37 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Using Pytorch implementation of U-Net architecture for road and building extraction

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published