FlowReg / README.md
mathpluscode's picture
Create README.md
8dc9ff4 verified
metadata
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.

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 for full usage and training instructions.