SA-UNet: Spatial Attention U-Net for Retinal Vessel Segmentation(ICPR 2020)
Overview
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.
Datasets
Data augmentation:
- 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.
Quick start
Training
Run Train_drive.py or Train_chase.py
Testing
Run Eval_drive.py or Eval_chase.py
Environments
Keras 2.3.1
Tensorflow==1.14.0
About Keras
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.
If you are inspired by our work, please cite these papers.
Structured dropout convolutional block
@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}
}
14 Oct 27, 2022
23 Aug 20, 2022
869 Jan 7, 2023
26 Dec 3, 2022
26 Aug 25, 2022
1 Oct 10, 2021
72 Dec 19, 2022
147 Dec 3, 2022
5 Nov 21, 2022
0 Sep 2, 2021
4 Dec 21, 2021
9 Dec 7, 2022
1 Dec 17, 2021
3 Jun 29, 2022
104 Nov 25, 2022
1.2k Jan 4, 2023
24 Oct 22, 2022
43 Oct 3, 2022
4 May 26, 2022