# Image-Registration-with-CNN

**Repository Path**: AI52CV/image-registration-with-cnn

## Basic Information

- **Project Name**: Image-Registration-with-CNN
- **Description**: This is the official Pytorch implementation of "Fast Symmetric Diffeomorphic Image Registration with Convolutional Neural Networks" (CVPR 2020), written by Tony C. W. Mok and Albert C. S. Chung.
代码原地址：https://github.com/cwmok/Fast-Symmetric-Diffeomorphic-Image-Registration-with-Convolutional-Neural-Networks
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 3
- **Forks**: 0
- **Created**: 2021-04-06
- **Last Updated**: 2024-03-28

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Fast Symmetric Diffeomorphic Image Registration with Convolutional Neural Networks

This is the official Pytorch implementation of "Fast Symmetric Diffeomorphic Image Registration with Convolutional Neural Networks" (CVPR 2020), written by Tony C. W. Mok and Albert C. S. Chung.

\*\* Please also check out our new deep learning-based image registration framework ([LapIRN - MICCAI2020](https://arxiv.org/abs/2006.16148 "eprint arXiv:2006.16148")) at https://github.com/cwmok/LapIRN, which achieved promising registration performance in large deformation settings. \*\*

## Prerequisites
- `Python 3.5.2+`
- `Pytorch 1.3.0`
- `NumPy`
- `NiBabel`

This code has been tested with `Pytorch` and GTX1080TI GPU.

## Inference
```
python Test_SYMNet.py
```

## Training
If you want to train a new model using your own dataset, please define your own data generator for `train_SYMNet.py` and perform the following script.

```
python train_sym_onepass.py
```

## Publication
If you find this repository useful, please cite:

- **Fast Symmetric Diffeomorphic Image Registration with Convolutional Neural Networks**  
[Tony C. W. Mok](https://cwmok.github.io/ "Tony C. W. Mok"), Albert C. S. Chung  
CVPR 2020. [eprint arXiv:2003.09514](https://arxiv.org/abs/2003.09514 "eprint arXiv:2003.09514")

## Notes on this repository
We found that estimating the time 1 solution instead of 0.5 solution tends to produce a smoother result in our later experiments. If you want to switch back to 0.5 solution, please replace the "self.time_step" with "self.time_step-1" at line 165 in `Models.py` and train a new model from scratch.

## Acknowledgment
Some codes in this repository are modified from [IC-Net](https://github.com/zhangjun001/ICNet) and [VoxelMorph](https://github.com/voxelmorph/voxelmorph).

###### Keywords
Keywords: Diffeomorphic Image Registration, convolutional neural networks, alignment, Symmetric Image Registration