File size: 3,238 Bytes
4ada69b 6837773 4ada69b a3049f0 4ada69b 05c5e2e 4ada69b 6837773 4ada69b 6837773 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 (ICCV 2025)
This repository contains the pre-trained checkpoints of MRGen: https://arxiv.org/abs/2412.04106/.
## Some Information
[Project Page](https://haoningwu3639.github.io/MRGen/) 路 [Paper](https://arxiv.org/abs/2412.04106/) 路 [Dataset](https://huggingface.co/datasets/haoningwu/MRGen-DB) 路 [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
Please check them out [here](https://huggingface.co/haoningwu/MRGen/tree/main/nnUNet_segmentation_model_zoo/CHAOS-MRI_T1_to_CHAOS-MRI_T2/nnUNetTrainer__nnUNetPlans__2d).
## Citation
If you use this code and data for your research or project, please cite:
@inproceedings{wu2025mrgen,
author = {Wu, Haoning and Zhao, Ziheng and Zhang, Ya and Wang, Yanfeng and Xie, Weidi},
title = {MRGen: Segmentation Data Engine for Underrepresented MRI Modalities},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)}
year = {2025},
}
## Contact
If you have any questions, please feel free to contact haoningwu3639@gmail.com or Zhao_Ziheng@sjtu.edu.cn. |