Official Tensorflow Implementation of "A Self-Adaptive and Multi-Attention Deep Convolutional Network for Retinal OCT Fluid Segmentation".Paper
- [Tensorflow] by AidiaLAB.
- Clone this repo:
git clone -b master --single-branch https://github.com/arminbiglari/RetiFluidNet.git
cd RetiFluidNet- Install Dependencies:
pip install -r requirments.txt- To train the model, Please adjust the params.yaml file based on your dataset.
python3 train.py- To test the model, Please adjust the params.yaml file based on your dataset.
python3 test.py- To access pre-traiend models please contact [email protected]
If you find this useful for your research, please use the following.
@ARTICLE{9980422,
author={Rasti, Reza and Biglari, Armin and Rezapourian, Mohammad and Yang, Ziyun and Farsiu, Sina},
journal={IEEE Transactions on Medical Imaging},
title={RetiFluidNet: A Self-Adaptive and Multi-Attention Deep Convolutional Network for Retinal OCT Fluid Segmentation},
year={2023},
volume={42},
number={5},
pages={1413-1423},
keywords={Fluids;Retina;Image segmentation;Visualization;Task analysis;Optimization;Lesions;Medical image segmentation;convolutional neural network;retinal disease;fluid segmentation},
doi={10.1109/TMI.2022.3228285}}