UNet for retina vessel segmentation.

We run the code on Ubuntu 18.04 LTS with a GTX 1080ti GPU.

Python (3.7.3) | Tensorflow (2.0.0) | CUDA (10.0) | cuDNN (7.6.0)

You can download the datasets from here. And then move the files into the ./data/ folder.

Because of my RAM limited, I firstly transform the train.mat file to .tfrecord files.

python preprocess.py

Then you can train the model initially.

CUDA_VISIBLE_DEVICES=0 python main.py -e train -c ./config/config_0.ini

When you have trained successfully, you can evaluate the model.

CUDA_VISIBLE_DEVICES=0 python main.py -e test -c ./config/config_0.ini

For threshold: 0.5

https://github.com/orobix/retina-unet