/SPnet

Primary LanguagePython

SPnet

This is the source code for our paper "Symmetric Pyramid Network for Medical Image Inverse Consistent Diffeomorphic Registration"

Requirements

  • pytorch >= 1.6.0
  • python >= 3.7
  • Some basic python packages such as Numpy, nibabel

Usage

  1. Train the model
    python train.py
  2. Test the model
    python test.py