/ScaMorph

ScaMorph: A Robust Unsupervised Learning Model for Deformable Image Registration using Scale-aware Context Aggregation

Primary LanguagePythonMIT LicenseMIT

ScaMorph: A Robust Unsupervised Learning Model for Deformable Image Registration using Scale-aware Context Aggregation

Keywords:Deep Learning, Deformable Image Registration, Convolutional Modulation, Scale-aware Context Aggregation

Training and Testing

Training

python train_ScaMorph.py --atlas_dir atlas_dir --train_dir train_dir --val_dir val_dir --gpu 0 
python train_ScaMorph_diff.py --atlas_dir atlas_dir --train_dir train_dir --val_dir val_dir --gpu 0 

Testing

python infer.py

Quantitative results

Reference:

VoxelMorph TransMorph LKU-Net SMT