isikdogan/deepriver

a deep-learning-based river centerline extraction model

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

deepriver's Issues

Does this project require a training set

Does this project require a training set

Needs update to Tensorflow 2

Tensorflow.contrib is gone in tf 2.0.0, and tensorflow_core.compat.v1.contrib does not exist. Please update, current code is unusable.