zeahub/augmented-camus-segmentation

If you use this model, you should cite the following work: Van De Vyver, Gilles, et al. "Generative augmentations for improved cardiac ultrasound segmentation using diffusion models." arXiv preprint arXiv:2502.20100 (2025).

The repo contains a nnU-Net model for cardiac segmentatoin on apical two and four chamber views. It segments the LV and Myocardium

This model is trained on the augmented CAMUS dataset. The augmentations are all generative augmentations from the work mentioned above.

This model uses the nnU-Net architecture: Isensee, F., Jaeger, P.F., Kohl, S.A.A. et al. nnU-Net: a self-configuring method for deep learning-based biomedical image segmentation. Nat Methods 18, 203–211 (2021). https://doi.org/10.1038/s41592-020-01008-z