Skip to content
/ SkelCon Public

PyTorch implementation for our paper on TMI2022: Retinal Vessel Segmentation with Skeletal Prior and Contrastive Loss

Notifications You must be signed in to change notification settings

tyb311/SkelCon

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SkelCon

PyTorch implementation for our paper on TMI2022:

"Retinal Vessel Segmentation with Skeletal Prior and Contrastive Loss"

WebPages:

Method

Project

Project for SkelCon
    ├── code:core code for contribution in our paper
        ├── color_space_mixture.py (Data Augmentation Method)  
        ├── sample_contrastive_learning.py (Sample Contrastive Learning)  
        ├── model_skelcon.py (Skeletal Prior based Network)  
        └── ...  
    ├── docs (figures)  
        └── ... 
    ├── onnx (trained weights)  
        ├── *.onnx (Pytorch trained weights)  
        ├── infer.py (to extract vessels from fundus images with *.onnx) 
        └── ...     
    ├── proj (package for segmentation with torch)  
        ├── data (to extract datasets)  
        ├── nets (define the network)  
        ├── build.py (define the network)  
        ├── grad.py (for training)  
        ├── loop.py (for training)  
        ├── optim.py (optimizer)  
        ├── main.py   
        └── ...  
    ├── results (segmentation for fundus images on testsets)  
        ├── popular (segmentation results for popular datasets)  
        ├── generalization (segmentation results for cross-dataset-validation)  
        └── ...   

And for the training on DRIVE dataset, run the command

cd proj
python main.py --gpu=1 --db=drive

Contact

For any questions, please contact me. And my e-mails are

Citation

If you use this codes in your research, please cite the paper:

@article{tan2022retinal,
  title={Retinal Vessel Segmentation with Skeletal Prior and Contrastive Loss},
  author={Tan, Yubo and Yang, Kai-Fu and Zhao, Shi-Xuan and Li, Yong-Jie},
  journal={IEEE Transactions on Medical Imaging},
  doi={10.1109/TMI.2022.3161681},
  volume    = {41},
  number    = {9},
  pages     = {2238--2251},
  year      = {2022},
  publisher={IEEE}
}

About

PyTorch implementation for our paper on TMI2022: Retinal Vessel Segmentation with Skeletal Prior and Contrastive Loss

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages