Deeply supervised network for white matter hyperintensities segmentation with transfer learning

Dependency installation

pip install -r requirements.txt

Pretrained weight

All pretrained weights can be downloaded from https://drive.google.com/drive/folders/1F_PXPuGTU76clUy8sWmKIH-HJyXYOOXh?usp=sharing.

To test your cases

python3 predict.py -i path/to/input_file -o path/to/output_file

To cite

wu2022deeply,
title={Deeply supervised network for white matter hyperintensities segmentation with transfer learning},
author={Yilei Wu and Fang Ji and Yao Feng Chong and Li-Hsian Christopher Chen and Juan Helen Zhou},
booktitle={Medical Imaging with Deep Learning},
year={2022},
url={https://openreview.net/forum?id=KbXNnCCXN-i}
}