carlesventura/iterative-deep-learning

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

iterative-deep-learning's Issues

results_test_vgg folder is missing

dear author, the folder of results_test_vgg in iterative_roads_local_mask.py is not provided, could you please upload it, or where can I download it?

cannot find results_dir= './results_dir_vessels/results_DRIU_vessel_segmentation/'

Hi carlesventura,
I tried to run the code vessels/iterative/iterative_graph_creation_no_mask_offset.py, and it reports an error: cannot find the results_dir = './results_dir_vessels/results_DRIU_vessel_segmentation/'.
I didn't find any message in your code about how to get 'results_dir',can you give me some help?
Thank you very much!

Generate gt value

Hi carlesventura,
Thank you for sharing. I am confused about generate_gt_val_roads.py. Should i create patchs for training data or just create them for validation data？ I will be very appreciate if you can share the directory figure to me.
Thank you very much!