| | --- |
| | license: mit |
| | library_name: pytorch |
| | tags: |
| | - medical-imaging |
| | - image-registration |
| | - cardiac-mri |
| | - flow-matching |
| | datasets: |
| | - mathpluscode/ACDC-FlowReg |
| | --- |
| | |
| | # FlowReg: Flow Matching-enabled Test-Time Refinement for Unsupervised Cardiac MR Registration |
| |
|
| | Pretrained checkpoints for **FlowReg** and **CorrMLP** for unsupervised deformable 3D medical image registration on cardiac MRI. |
| |
|
| | - **Paper**: [Flow Matching-enabled Test-Time Refinement for Unsupervised Cardiac MR Registration](https://arxiv.org/abs/2603.01073) |
| | - **Code**: [GitHub](https://github.com/mathpluscode/FlowReg) |
| | - **Dataset**: [mathpluscode/ACDC-FlowReg](https://huggingface.co/datasets/mathpluscode/ACDC-FlowReg) |
| |
|
| | ## Available Checkpoints |
| |
|
| | | Model | Dataset | Direction | Subfolder | |
| | |---------|---------|-----------|------------------------| |
| | | FlowReg | ACDC | ED→ES | `flowreg/acdc_ed2es` | |
| | | FlowReg | ACDC | ES→ED | `flowreg/acdc_es2ed` | |
| | | FlowReg | M&Ms-2 | ED→ES | `flowreg/mnms2_ed2es` | |
| | | FlowReg | M&Ms-2 | ES→ED | `flowreg/mnms2_es2ed` | |
| | | CorrMLP | ACDC | ED→ES | `corrmlp/acdc_ed2es` | |
| | | CorrMLP | ACDC | ES→ED | `corrmlp/acdc_es2ed` | |
| | | CorrMLP | M&Ms-2 | ED→ES | `corrmlp/mnms2_ed2es` | |
| | | CorrMLP | M&Ms-2 | ES→ED | `corrmlp/mnms2_es2ed` | |
| |
|
| | See the [GitHub repo](https://github.com/mathpluscode/FlowReg) for full usage and training instructions. |
| |
|
