This code is for the paper: Spatial Attention U-Net for Retinal Vessel Segmentation. We report state-of-the-art performances on DRIVE and CHASE DB1 datasets.
Code written by Changlu Guo, Budapest University of Technology and Economics(BME).
We train and evaluate on Ubuntu 16.04, it will also work for Windows and OS.
- keras_dataAug.py
(1) Random rotation;
(2) adding Gaussian noise;
(3) color jittering;
2.flip.py
(4) horizontal, vertical and diagonal flips.
if you do not want to do above augmentation,just download it from my link.
Run Train_drive.py or Train_chase.py
Run Eval_drive.py or Eval_chase.py
Keras 2.3.1
Tensorflow==1.14.0
Keras is a minimalist, highly modular neural networks library, written in Python and capable of running on top of either TensorFlow or Theano. It was developed with a focus on enabling fast experimentation. Being able to go from idea to result with the least possible delay is key to doing good research.
Use Keras if you need a deep learning library that:
allows for easy and fast prototyping (through total modularity, minimalism, and extensibility). supports both convolutional networks and recurrent networks, as well as combinations of the two. supports arbitrary connectivity schemes (including multi-input and multi-output training). runs seamlessly on CPU and GPU. Read the documentation Keras.io
Keras is compatible with: Python 2.7-3.5.
@INPROCEEDINGS{8942005,
author={C. {Guo} and M. {Szemenyei} and Y. {Pei} and Y. {Yi} and W. {Zhou}},
booktitle={2019 IEEE 19th International Conference on Bioinformatics and Bioengineering (BIBE)},
title={SD-Unet: A Structured Dropout U-Net for Retinal Vessel Segmentation},
year={2019},
volume={},
number={},
pages={439-444},}
@article{Guo2020SAUNetSA,
title={SA-UNet: Spatial Attention U-Net for Retinal Vessel Segmentation},
author={Changlu Guo and Marton Szemenyei and Yugen Yi and Wenle Wang and Buer Chen and Changqi Fan},
journal={ArXiv},
year={2020},
volume={abs/2004.03696}
}