Skip to content
master
Switch branches/tags
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

README.md

Iterative Deep Learning for Road Topology Extraction

Published at BMVC 2018.

Paper available on ArXiv: https://arxiv.org/abs/1808.09814

Paper website: https://carlesventura.github.io/iterative-dl-road-website/

Code instructions

Downloading datasets:

Download Massachusetts Roads Dataset from website: https://www.cs.toronto.edu/~vmnih/data/

Download DRIVE dataset (vessels from retina images) from website: https://www.isi.uu.nl/Research/Databases/DRIVE/

Download graph annotations for DRIVE dataset from website: http://people.duke.edu/~sf59/Estrada_TMI_2015_dataset.htm

Set your work directory, create a directory inside named gt_dbs and copy there the downloaded datasets (roads dataset in a folder named MassachusettsRoads, DRIVE dataset in a folder named DRIVE and graph annotations for DRIVE in a folder named artery-vein).

Experiments for road topology extraction:

  1. Generate road patches for training the patch-level model: roads/patch/generate_gt_val_roads.py
  2. Train patch-level model: roads/patch/train_road_patches.py
  3. (Optional) Evaluate patch-level model: roads/patch/evaluation/PR_evaluation_patch_roads.py
  4. Apply the patch-level model iteratively over the road test images: roads/iterative/iterative_roads_local_mask.py
  5. (Optional) Evaluate iterative results: roads/iterative/evaluation/connectivity_evaluation_roads.py

Experiments for vessel topology extraction:

  1. Train patch-level model: vessels/patch/train_hg.py
  2. (Optional) Evaluate patch-level model: vessels/patch/evaluation/PR_evaluation.py
  3. Apply the patch-level model iteratively over the retina test images: vessels/iterative/iterative_graph_creation_no_mask_offset.py
  4. (Optional) Evaluate iterative results: vessels/iterative/evaluation/connectivity_evaluation.py

About

No description, website, or topics provided.

Resources

Releases

No releases published

Packages

No packages published

Languages