```bash
├── data_preparation
│   ├── main_preperation.py  (load .nii files, perform motion simulation and save them to .h5 file)
│   ├── motion_simulation2D.py
├── datasets
│   ├── data_loader.py
├── losses
│   ├── losses.py (contain DC and Dice loss functions)
├── networks
│   ├── seg_net.py (motion detection network)
│   ├── unet_swin.py (motion correction network)
│   ├── swin_transformer.py
│   ├── fp16_util.py
│   ├── basic_ops.py
├── build.py
├── main.py
└── utils.py

@misc{safari2025physicsinformeddeeplearningmodel,
      title={A Physics-Informed Deep Learning Model for MRI Brain Motion Correction}, 
      author={Mojtaba Safari and Shansong Wang and Zach Eidex and Richard Qiu and Chih-Wei Chang and David S. Yu and Xiaofeng Yang},
      year={2025},
      eprint={2502.09296},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2502.09296}, 
}