File size: 1,429 Bytes
8dc9ff4 | 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 | ---
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.
|