Improve model card with metadata and details

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +13 -2
README.md CHANGED
@@ -1,7 +1,18 @@
1
  ---
2
- license: cc-by-4.0
 
 
3
  ---
 
 
 
4
  You should cite the following paper when using the code in this repository:
5
 
6
  Van De Vyver, Gilles, et al. "Generative augmentations for improved cardiac ultrasound segmentation using diffusion models." arXiv preprint arXiv:2502.20100 (2025).
7
- https://arxiv.org/abs/2502.20100
 
 
 
 
 
 
 
1
  ---
2
+ license: cc-by-nc-sa-4.0
3
+ library_name: diffusers
4
+ pipeline_tag: image-to-image
5
  ---
6
+
7
+ This model is a diffusion model trained to generate augmentations for cardiac ultrasound images, as described in [Generative augmentations for improved cardiac ultrasound segmentation using diffusion models](https://hf.co/papers/2502.20100). These augmentations improve the robustness of segmentation models trained on cardiac ultrasound data. The model is designed to increase the diversity of training data, leading to better generalization to unseen datasets.
8
+
9
  You should cite the following paper when using the code in this repository:
10
 
11
  Van De Vyver, Gilles, et al. "Generative augmentations for improved cardiac ultrasound segmentation using diffusion models." arXiv preprint arXiv:2502.20100 (2025).
12
+ https://arxiv.org/abs/2502.20100
13
+
14
+ Blog post: [https://gillesvandevyver.com/#/projects/generative-ai](https://gillesvandevyver.com/#/projects/generative-ai)
15
+
16
+ Code: https://github.com/GillesVanDeVyver/EchoGAINS
17
+
18
+ A Jupyter notebook tutorial is available at [tutorial/tutorial.ipynb](tutorial/tutorial.ipynb).