Wavelet transform and edge loss-based three-stage segmentation model for retinal vessel

Descriptions

This repository contains the implementations of the paper 'Wavelet transform and edge loss-based three-stage segmentation model for retinal vessel' published in the Biomedical Signal Processing and Control journal.

This repository currently provides examples of implementing, training, and inferring with the core model code. It also includes guidance on running the code on sample datasets. Additionally, there are suggestions and code for visual analysis.

[Get Paper]

🔑 Steps for Training and Testing

⚡ pip install -r requirements.txt

🐍 run python main.py

❗ Note❗

Remeber delete the .gitkeep and .DS_Store files before running this program.

Citation

If you find this code is useful in your research, please consider to cite:

@article{li2023wavelet,
  title={Wavelet transform and edge loss-based three-stage segmentation model for retinal vessel},
  author={Li, Xuecheng and Zheng, Yuanjie and Zang, Mengwei and Jiao, Wanzhen},
  journal={Biomedical Signal Processing and Control},
  volume={86},
  pages={105355},
  year={2023},
  publisher={Elsevier}
}