You should cite the following paper when using the code in this repository:

Van De Vyver, Gilles, et al. "Generative augmentations for improved cardiac ultrasound segmentation using diffusion models." arXiv preprint arXiv:2502.20100 (2025). https://arxiv.org/abs/2502.20100

To use the model, unzip nnUNetTrainer__nnUNetPlans__2d.zip and follow the instuctions in inference_instructions.txt. For more information on the model, see the documentation of nnU-Net.

This nnU-Net model is for cardiac segmentatoin on apical two and four chamber views.

This model is trained on an augmented version of the CAMUS dataset: S. Leclerc, E. Smistad, J. Pedrosa, A. Ostvik, et al. "Deep Learning for Segmentation using an Open Large-Scale Dataset in 2D Echocardiography" in IEEE Transactions on Medical Imaging, vol. 38, no. 9, pp. 2198-2210, Sept. 2019. doi: 10.1109/TMI.2019.2900516

Code and nformation about the augmentations can be found here: https://github.com/GillesVanDeVyver/EchoGAINS

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

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.