Paired lung CT registration¶
Note: Please read the DeepReg Demo Disclaimer.
Author¶
DeepReg Development Team (Shaheer Saeed)
Application¶
This is a registration between CT images acquired at different time points for a single patient. The images being registered are taken at inspiration and expiration for each subject. This is an intra subject registration. This type of intra subject registration is useful when there is a need to track certain features on a medical image such as tumor location when conducting invasive procedures.
Instruction¶
Change current directory to the root directory of DeepReg project;
The
demo_data.py
,demo_train.py
anddemo_predict.py
scripts need to be run using the following command:
python3 demos/paired_ct_lung/script_name.py
A short description of the scripts is provided below. The scripts must be run in the following order:
Run the
demo_data.py
script: This script does the following:Download data using
tf.keras.utils.get_file
. Data is downloaded to the demo directory but this can be changed (instructions in the comments in the script).Split the data into three sets train, valid and test (change ratio_of_test_and_valid_samples variable to change the ratio of test and valid samples)
Restructure the files, for each of the train, valid and test sets, into a directory structure that is suitable for use with the paired loader in DeepReg
Rescale all images to 0-255 so they are suitable for use with DeepReg
Download a pretrained model to use with the predict script
Run the
demo_train.py
script: This script does the following:Specify the training options like GPU support
Specify the config file paths (the config file to define the network is one which is avialable with DeepReg and the config file for the data is given in the demo folder)
Train a network using DeepReg
Run the
demo_predict.py
script: This script does the following:Use the pretrained network to make predictions for the test set
Use the predictions to plot the results (the path to the images generated in the logs will need to be specified, look at comments in the script to change this)
Pre-trained Model¶
A pre-trained model will be downloaded after running demo_data.py
and unzipped at the
dataset folder under the demo folder. This pre-trained model will be used by default
with deepreg_predict
. Run the user-trained model by specifying with --ckpt_path
the
location where the ckpt files will be saved, in this case (specified by deepreg_train
as above), /logs/learn2reg_t2_paired_train_logs/.
Data¶
The dataset for this demo comes from Lean2Reg Challenge: CT Lung Registration - Training Data [1].
Tested DeepReg version¶
Last commit at which demo was tested: v. 0.1.6-alpha
Contact¶
Please raise an issue.
Reference¶
[1] Hering, Alessa, Murphy,Keelin, and van Ginneken, Bram. (2020). Lean2Reg Challenge: CT Lung Registration : CT Lung Registration - Training Data