Update README.md

README.md

@@ -1,3 +1,22 @@
 ---
 license: 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 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/listSubGroups
+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_camus_cv1