File size: 3,053 Bytes
4ada69b 8c09d31 4ada69b 8c09d31 4ada69b 05c5e2e 4ada69b 8c09d31 4ada69b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
---
license: mit
---
# MRGen: Segmentation Data Engine for Underrepresented MRI 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/MRGen-DB) $\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 |
|:-----------------:|:-----------------:|:------:|
| CHAOS-MRI_T1 | CHAOS-MRI_T2-SPIR | [CHAOS-MRI_T1_to_CHAOS-MRI_T2](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_CHAOS-MRI_T1_to_CHAOS-MRI_T2) |
| CHAOS-MRI_T2-SPIR | CHAOS-MRI_T1 | [CHAOS-MRI_T2_to_CHAOS-MRI_T1](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_CHAOS-MRI_T2_to_CHAOS-MRI_T1) |
| MSD-Prostate_T2 | MSD-Prostate_ADC | [MSDProstate_T2_to_MSDProstate_ADC](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_MSDProstate_T2_to_MSDProstate_ADC) |
| MSD-Prostate_ADC | MSD-Prostate_T2 | [MSDProstate_ADC_to_MSDProstate_T2](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_MSDProstate_ADC_to_MSDProstate_T2) |
| PanSeg_T1 | PanSeg_T2 | [PanSeg_T1_to_PanSeg_T2](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_PanSeg_T1_to_PanSeg_T2) |
| PanSeg_T2 | PanSeg_T1 | [PanSeg_T2_to_PanSeg_T1](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_PanSeg_T2_to_PanSeg_T1) |
| LiQA_T1 | CHAOS-MRI_T2-SPIR | [LiQA_T1_to_CHAOS-MRI_T2](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_LiQA_T1_to_CHAOS-MRI_T2) |
| CHAOS-MRI_T2-SPIR | LiQA_T1 | [CHAOS-MRI_T2_to_LiQA_T1](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_CHAOS-MRI_T2_to_LiQA_T1) |
| MSD-Prostate_ADC | PROMISE12_T2 | [MSDProstate_ADC_to_PROMISE12_T2](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_MSDProstate_ADC_to_PROMISE12_T2) |
| PROMISE12_T2 | MSD-Prostate_ADC | [PROMISE12_T2_to_MSDProstate_ADC](https://huggingface.co/haoningwu/MRGen/tree/main/controlnet_model_zoo/controlnet_PROMISE12_T2_to_MSDProstate_ADC) |
### 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: Segmentation Data Engine for Underrepresented MRI Modalities},
journal = {arXiv preprint arXiv:2412.04106},
year = {2024},
}
## Contact
If you have any questions, please feel free to contact [email protected] or Zhao[email protected]. |