Official PyTorch implementation of the SPAC "Stochastic Planner-Actor-Critic for Unsupervised Deformable Image Registration", AAAI 2022. ![visitors](https://camo.githubusercontent.com/08833bf3b0afce7637e76e87f8995e0344cc7f737148f08ceecc2c717ce6a709/68747470733a2f2f76697369746f722d62616467652e676c697463682e6d652f62616467653f706167655f69643d416c676f6c7a772f535041432d4465666f726d61626c652d526567697374726174696f6e)
-
Version
- python 3.6
- pytorch 1.4
-
Install requirements:
pip install -r requirements.txt
- The core algorithm is in brain.py, and we build environment in env.py. The agent.py handles the logic of interactions.
- Modify config.py to set path to data, iterations, and other parameters.
- To train the model(s) in the paper, run this command:
python main.py
- Set 'idx' parameter in config.py to choose existing models.
- To test our model, run:
python test.py
If our code helps your research or work, please consider citing our paper.
@inproceedings{luo2022stochastic,
title={Stochastic Planner-Actor-Critic for Unsupervised Deformable Image Registration},
author={Luo, Ziwei and Hu, Jing and Wang, Xin and Hu, Shu and Kong, Bin and Yin, Youbing and Song, Qi and Wu, Xi and Lyu, Siwei},
booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
volume={36},
number={2},
pages={1917--1925},
year={2022}
}
@inproceedings{Luo2021StochasticAF,
title={Stochastic Actor-Executor-Critic for Image-to-Image Translation},
author={Ziwei Luo and Jing Hu and Xin Wang and Siwei Lyu and Bin Kong and Youbing Yin and Qi Song and Xi Wu},
booktitle={IJCAI},
year={2021}
}
If you have any question, contact me: [email protected]