Improve model card with metadata and details
Browse filesThis PR improves the model card by:
- Adding the `pipeline_tag: image-to-image` to clarify the model's function.
- Specifying the `library_name: diffusers` based on the model's use of diffusion models.
- Correcting the license to `cc-by-nc-sa-4.0`.
- Adding a more descriptive model description.
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).
|