Skip to content
/ WS-DMF Public

W-shaped deep matched filtering for retinal vessel segmentation

Notifications You must be signed in to change notification settings

tyb311/WS-DMF

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

WS-DMF

PyTorch implementation for our paper (submitted):

"Deep matched filtering for retinal vessel segmentation"

The full material will be made available once the manuscript is accepted!!!

Project

Project for WS-DMF
    ├── data 
        ├── eyeset.py (dataset & dataloader & data pre-processing)  
        └── ...  
    ├── nets 
        ├── modules/activation.py (activation functions)  
        ├── modules/attention.py (attention modules)  
        ├── conv.py (convolution layers)  
        ├── dmfu.py (deep matched filtering)  
        ├── lunet.py (lightweight UNet)  
        ├── rot.py (APC layer related OAL loss function)  
        └── ...   
    ├── utils  
        ├── loss.py (loss function)  
        ├── optim.py (optimizer for training)  
        └── ...  
    ├── build.py (implementation for WS-DMF)  
    ├── grad.py (implementation for backgrading)  
    ├── loop.py (implementation for training)  
    ├── main.py (implementation for main function)  
    └── ...   
    ├── onnx (Pytorch trained weights)  
        └── infer.py (infer fundus images for segmentation with *.onnx weights)  
        └── ...  
    ├── results (segmentation results)  
        └── ...  (segmentation results for popular datasets)
        └── ...  (segmentation results for cross-dataset-validation)

Contact

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

About

W-shaped deep matched filtering for retinal vessel segmentation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages