Update README.md

README.md

@@ -1,3 +1,39 @@
----
-license: mit
----
+---
+license: mit
+---
+
+# MRGen: Diffusion-based Controllable Data Engine for MRI Segmentation towards Unannotated Modalities
+This repository contains the pre-trained checkpoints of MRGen: https://arxiv.org/abs/2412.04106/.
+
+## Some Information
+[Project Page](https://haoningwu3639.github.io/MRGen/) $\cdot$ [Paper](https://arxiv.org/abs/2412.04106/) $\cdot$ [Dataset](https://huggingface.co/datasets/haoningwu/MedGen-1M) $\cdot$ [Checkpoints](https://huggingface.co/haoningwu/MRGen)
+
+## Checkpoints - Model Zoo
+Please refer to [MRGen](https://huggingface.co/haoningwu/MRGen) to download our pre-trained checkpoints.
+
+### Generative Model Checkpoints
+
+| Source Modality | Target Modality | Link |
+|-----------------|-----------------|------|
+|       |       |
+|       |       |
+
+### Segmentation Model Checkpoints
+To be updated soon...
+
+## Citation
+If you use this code and data for your research or project, please cite:
+
+	@misc{wu2024mrgen,
+            author  = {Wu, Haoning and Zhao, Ziheng and Zhang, Ya and Xie, Weidi and Wang, Yanfeng},
+            title   = {MRGen: Diffusion-based Controllable Data Engine for MRI Segmentation towards Unannotated Modalities},
+            journal = {arXiv preprint arXiv:2412.04106},
+            year    = {2024},
+      }
+
+## Acknowledgements
+Many thanks to the code bases from [diffusers](https://github.com/huggingface/diffusers) and [SimpleSDM](https://github.com/haoningwu3639/SimpleSDM).
+
+
+## Contact
+If you have any questions, please feel free to contact [email protected] or [email protected].