markcunningham0410 / shape-attentive-unet Goto Github PK

Code for our paper "SAUNet: Shape Attentive U-Net for Interpretable Medical Image Segmentation". https://arxiv.org/pdf/2001.07645.pdf

(January 21, 2020) Notice: we are currently working on cleaning up the code and fixing instabilities. We will provide further documentation and updates here in the near future.

(January 29, 2020) We have fixed major instabilities and cleaned up most of the code. Feel free to clone and contact us if there are any bugs.

If you find our work helpful, please consider citing our work:

@misc{sun2020saunet,
    title={SAUNet: Shape Attentive U-Net for Interpretable Medical Image Segmentation},
    author={Jesse Sun and Fatemeh Darbeha and Mark Zaidi and Bo Wang},
    year={2020},
    eprint={2001.07645},
    archivePrefix={arXiv},
    primaryClass={eess.IV}
}