Downloads · 30 days
0
gillesvdv/nnunet_camus_cv1
nnunet_camus_cv1 is a machine learning model from gillesvdv. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as cc-by-4.0.
This model is trained on 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 Medic…
Downloads · 30 days
0
Access
Public
Updated Dec 1, 2023
Repo size
189 MB
Likes
0
Public
Click a slice to open those files.
.pth189 MB · 99%
From the Hugging Face model README
This model is trained on 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
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
This model is part of the following work. You must cite this paper for any use of the model: G.Van De Vyver, S. Thomas, G. Ben-Yosef, S. H. Olaisen, H. Dalen, L. Løvstakken, and E. Smistad: “Towards Robust Cardiac Segmentation using Graph Convolutional Networks” in arXiv preprint arXiv:2310.01210, 2023, https://github.com/gillesvntnu/GCN_multistructure.git
The model is the .pth version of the nnU-Net model as described in the work mentioned above. It is trained on the first split of the CAMUS dataset.
The model can be directly used in the framework provided at https://github.com/gillesvntnu/GCN_multistructure.git
Information on the splits of the CAMUS dataset can be found at: https://github.com/gillesvntnu/GCN_multistructure/tree/main/files/subgroups_CAMUS
For the corresponding version in .onnx format, see https://huggingface.co/gillesvdv/nnunet_camus_cv1_onnx