Official implementations of paper: Learning Euler's Elastica Model for Medical Image Segmentation, and a short version was accepted by ISBI 2021 .
If you want to use these methods just as constrains (combining with dice loss or ce loss), you can use torch.mean() to replace torch.sum().
Some important required packages include:
Follow official guidance to install. Pytorch.
If you find Active Contour Based Loss Functions are useful in your research, please consider to cite:
@inproceedings{chen2020aceloss,
title={Learning Euler's Elastica Model for Medical Image Segmentation},
author={Chen, Xu and Luo, Xiangde and Zhao, Yitian and Zhang, Shaoting and Wang, Guotai and Zheng, Yalin},
journal={arXiv preprint arXiv:2011.00526},
year={2020}
}
@inproceedings{chen2019learning,
title={Learning Active Contour Models for Medical Image Segmentation},
author={Chen, Xu and Williams, Bryan M and Vallabhaneni, Srinivasa R and Czanner, Gabriela and Williams, Rachel and Zheng, Yalin},
booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
pages={11632--11640},
year={2019}
}