---
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.