Skip to content
/ HED-UNet Public

Simultaneous Semantic Segmentation and Edge Detection

License

Notifications You must be signed in to change notification settings

khdlr/HED-UNet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HED-UNet

Code for HED-UNet, a model for simultaneous semantic segmentation and edge detection.

Glacier Fronts

This model was originally developed to detect calving front margins in Antarctica from Sentinel-1 SAR imagery.

glaciers

Building Footprints

As the original dataset isn't publicly available, this repository contains an adaption of the model for building footprint extraction on the Inria Aerial Image Labeling dataset. Here are some example results:

Bloomington

bloomington27-overview-binary

Innsbruck

innsbruck20-overview-binary

San Francisco

sfo20-overview-binary

Usage

In order to use this for your project, you will need adapt either the get_dataloader function in train.py or the methods in data_loading.py.

Citation

If you find our code helpful and use it in your research, please use the following BibTeX entry.

@article{HEDUNet2021,
  author={Heidler, Konrad and Mou, Lichao and Baumhoer, Celia and Dietz, Andreas and Zhu, Xiao Xiang},
  journal={IEEE Transactions on Geoscience and Remote Sensing}, 
  title={HED-UNet: Combined Segmentation and Edge Detection for Monitoring the Antarctic Coastline}, 
  year={2021},
  volume={},
  number={},
  pages={1-14},
  doi={10.1109/TGRS.2021.3064606}
}

About

Simultaneous Semantic Segmentation and Edge Detection

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages