Update README.md

README.md

@@ -1,3 +1,11 @@
----
-license: cc-by-4.0
----
+---
+license: cc-by-4.0
+---
+If you use this model, you should cite the following work:  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
+
+Multi structure graph convolutional network (GCN) for cardiac segmentatoin on apical two and four chamber views, trained on the fist cross validation split of the CAMUS dataset. See https://github.com/GillesVanDeVyver/arqee for usage.
+
+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
+
+
+The model is the .onnx version of the nnU-Net model as described in the first work mentioned above. It is trained and validated using first cross validation split of the CAMUS dataset. See https://github.com/gillesvntnu/GCN_multistructure.git for more details