gillesvdv/

GCN_with_displacement_camus_cv1

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 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 "GCN with displacement method" described in the work mentioned above. It is trained on the first split of the CAMUS dataset.

Information on the splits of the CAMUS dataset can be found at: https://github.com/gillesvntnu/GCN_multistructure/tree/main/files/subgroups_CAMUS

The model can be directly used in the framework provided at https://github.com/gillesvntnu/GCN_multistructure.git

For the corresponding version in .onnx format, see https://huggingface.co/gillesvdv/GCN_with_displacement_camus_cv1_onnx