This code is for the paper: Channel Attention Residual U-Net for Retinal Vessel Segmentation. We report state-of-the-art performances on DRIVE, CHASE DB1 and STARE 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.
In this paper we propose a Modified Efficient Channel Attention (MECA), and its specific structure is shown in the figure below. The code can be found at attention_module.py
Train:
Run train_drive.py or train_chase.py
or tarin_stare.py(4-fold cross-validation)
Test: Run eval_drive.py or eval_chase.py
Keras 2.3.1
Tensorflow==1.14.0
@misc{guo2020channel,
title={Channel Attention Residual U-Net for Retinal Vessel Segmentation},
author={Changlu Guo and Márton Szemenyei and Yangtao Hu and Wenle Wang and Wei Zhou and Yugen Yi},
year={2020},
eprint={2004.03702},
archivePrefix={arXiv},
primaryClass={eess.IV}
}