https://github.com/zhangjun001/ICNet

Inverse-Consistent Deep Networks for Unsupervised Deformable Image Registration
https://github.com/zhangjun001/ICNet

Last synced: 20 days ago
JSON representation

Inverse-Consistent Deep Networks for Unsupervised Deformable Image Registration

• Host: GitHub
• URL: https://github.com/zhangjun001/ICNet
• Owner: zhangjun001
• License: mit
• Created: 2020-03-09T02:18:57.000Z (over 4 years ago)
• Default Branch: master
• Last Pushed: 2020-12-08T11:55:23.000Z (over 3 years ago)
• Last Synced: 2024-02-27T07:32:47.525Z (4 months ago)
• Language: Python
• Homepage:
• Size: 6.63 MB
• Stars: 52
• Watchers: 2
• Forks: 16
• Open Issues: 6
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Lists

• awesome-image-registration - Pytorch
• awesome_medical - ICNet

README

        
# Inverse-Consistent Deep Networks for Unsupervised Deformable Image Registration

Pytorch implementation for Inverse-Consistent Deep Networks for Unsupervised Deformable Image Registration.

This package includes 3D deformable image registration tool for brain images. The code was written by Jun Zhang, Tencent AI Healthcare.

# Applications

3D inverse-consistent deformable image registration

# Prerequisites

Linux python 2.7 NVIDIA GPU + CUDA CuDNN

Install pytorch (0.4.0 or 0.4.1) and dependencies from http://pytorch.org/ 

Install SimpleITK with `pip install SimpleITK` 

Install numpy with `pip install numpy`

# Apply

```

cd Code

```

Apply our Pre-trained Model (Note that the image pairs must be linearly aligned before using our code).

It is better to perform the histogram matching according to any one of our provided images. 

```

python Test.py --fixed ../Dataset/image_A.nii.gz --moving ../Dataset/image_B.nii.gz

```

# Train 

If you want to train a model using your own dataset, please perform the following script.

```

python Train.py --datapath yourdatapath

```

You need to perform the histogram matching for you dataset since we emply the `MSE loss` for measuring similarity.

You may need to adjust the parameters of `--inverse`, `--antifold`, and  `--smooth` for your dataset to get better registration performance.