a deep-learning-based river centerline extraction model
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
checkpoints
data
.gitignore
README.md
datagenerator.py
inference.py
nets.py
trainer.py

README.md

DeepRiver

A River Centerline Extraction Model that uses Deep Convolutional Neural Networks

Related papers

  • F. Isikdogan, A.C. Bovik, and P. Passalacqua, "Learning a River Network Extractor using an Adaptive Loss Function," IEEE Geoscience and Remote Sensing Letters, 2018. [Read at IEEExplore], [PDF]
  • F. Isikdogan, A.C. Bovik, and P. Passalacqua, "Surface Water Mapping by Deep Learning," IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2017. [Read at IEEExplore], [PDF]

Dependencies

  • Python 3.5+
  • OpenCV 3.0+
  • TensorFlow 1.4.0+
  • Numpy

Running instructions

Training a model from scratch:

python trainer.py

Running inference on a given image (extracting the centerlines from a surface water map):

python inference.py

Example images can be found in:

./data