This repository is the official PyTorch implementation of Low-light OCTA Images Segmentation Under the Constraints of Vascular Network Topology.

🚀 🚀 🚀 News:

2023/10/27: We released the source code

Optical Coherence Tomography Angiography (OCTA) provides unprecedented insights into the retinal vascular system. However, precise segmentation of OCTA images poses formidable challenges. Current methods contend with issues such as imaging artifacts and poor clarity under dark scene conditions that complicate the diagnosis and classification of diseases like Branch Vein Occlusion (BVO). To tackle these obstacles, we have developed a network based on topological structure generation that employs a transition from superficial to deep retinal layers, aiming to enhance OCTA segmentation accuracy. Both qualitative visual comparisons and quantitative metric analyses support improved performance. Moreover, our introduced enhancement methodology mitigates artifacts induced by low-light OCTA, reducing noise and improving clarity. Additionally, our system provides a structured approach to classifying BVO diseases, filling a significant gap in this domain. The overarching goal of these methods is to improve OCTA image quality and strengthen the reliability of segmentation. Preliminary assessments indicate potential for establishing robust fine-grained standards for OCTA vascular segmentation and analysis.

conda create --name bistim python=3.9 -y
conda activate bistim
conda install pytorch==1.10.0 torchvision==0.11.0 cudatoolkit=11.1 -c pytorch -c nvidia
pip install opencv-python pycocotools matplotlib onnxruntime onnx timm

First download a model checkpoint. Then the model can be used in just a few lines to get masks from a given prompt:

from models import bistim
input = torch.randn((1, 3, 512, 512))
model = bistim(num_classes=3)
out = model(input)

To train bistim:

python train.py

If you find BISTimuseful in your research or refer to the provided baseline results, please star ⭐ this repository and consider citing 📝