This code is for the paper: Residual Spatial Attention Network 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.
Spatial Attention (SA) was introduced as a part of the convolutional block attention module for classification and detection. SA employs the inter-spatial relationship between features to produce a spatial attention map, the code can be found in attention_module.py.
In this paper, we proposed RSAB, because it integrates DropBlock to prevent overfitting and the advantages of spatial attention, we believe it has great potential in the field of medical image processing for building deep networks even for processing small data sets .
Train: Run train_drive.py or train_chase.py
Test: Run eval_drive.py or eval_chase.py
Row 1 is for DRIVE dataset. Row 2 is for CHASE DB1 dataset. (a) Color fundus images, (b) segmentation results of Backbone, (c) segmentation results of Backbone+DropBlock, (d) segmentation results of RSAN, (e) corresponding ground truths.
Keras 2.3.1
Tensorflow==1.14.0
@misc{guo2020residual,
title={Residual Spatial Attention Network for Retinal Vessel Segmentation},
author={Changlu Guo and Márton Szemenyei and Yugen Yi and Wei Zhou and Haodong Bian},
year={2020},
eprint={2009.08829},
archivePrefix={arXiv},
primaryClass={eess.IV}
}