Create README.md

README.md

@@ -0,0 +1,34 @@
+---
+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.