cyfeng16 / ssvs

Project page of the paper 'Self-Supervised Vessel Segmentation via Adversarial Learning'

Self-Supervised Vessel Segmentation via Adversarial Learning

If you find our dataset, code or paper useful, please cite as

@inproceedings{SSVS,
    title = {Self-Supervised Vessel Segmentation via Adversarial Learning},
    author = {Yuxin Ma, Yang Hua, Hanming Deng, Tao Song, Hao Wang, Zhengui Xue,Heng Cao, Ruhui Ma and Haibing Guan},
    booktitle = {IEEE International Conference on Computer Vision (ICCV)},
    year = {2021}
}

XCAD dataset

You can download the XCAD dataset from Dropbox or BaiduNetdisk with the key neia.

Code

Environment

Follow ssv_env.yml to create the environment.

Data Preparation

Put the downloaded dataset to dataset/ssv.

XCAD/train/* ->datasets/ssv/*
XCAD/test/images -> datasets/ssv/testB
XCAD/test/masks -> datasets/ssv/testA

Put some background images (e.g.,XCAD/train/trainC) to datasets/ssv/testC. Using fractal.py to synthesis fractals and put it in datasets/ssv/trainA.

Train

Follow the scripts of arun_train.sh to train the model.

Test and Evaluation

Use arun_test.sh for inference, and use eval/test.sh to get evaluation results.