SAGAN
Update 2019.01.22
For those who want to use the piglet dataset for CT denoising research and use this work as a baseline, please refer to this issue for details on how I used it.
Update 2018.03.27
The piglet dataset we used in the publication is now open for download! Please find the link in my personal webpage. (Note: for non-commercial use only)
This repo provides the trained denoising model and testing code for low dose CT denoising as described in our paper.
Here are some randomly picked denoised results on low dose CTs from this kaggle challenge.
How to use
To better use this repo, please make sure the dose level of the LDCTs are larger than 0.71 mSv.
Prerequistites
- Linux or OSX
- NVIDIA GPU
- Python 3.x
- Torch7
Getting Started
- Install Torch7
- Install torch packages nngraph and hdf5
luarocks install nngraph
luarocks install hdf5
- Install Python 3.x (recommend using Anaconda)
- Install python dependencies
pip install -r requirements.txt
git clone [email protected]:xinario/SAGAN.git
cd SAGAN
#make a directory inside the root SAGAN folder to store your raw dicoms, e.g. ./dicoms
mkdir dicoms
#then put all your low dose CT images of dicom format into this folder and run
python pre_process.py --input ./dicoms --output ./datasets/experiment/test
#all your test images would now be saved as uint16 png format inside folder ./datasets/experiment/test.
DATA_ROOT=./datasets/experiment name=SAGAN which_direction=AtoB phase=test th test.lua
#the results are saved in ./result/SAGAN/latest_net_G_test/result.h5
- Display the result with a specific window, e.g. abdomen. Window type can be changed to ‘abdomen’, ‘bone’ or ‘none’
python post_process.py --window 'abdomen'
Now you can view the result by open the html file index.html sitting in the root folder
Citations
If you find it useful and are using the code/model/dataset provided here in a publication, please cite our paper:
Yi, X. & Babyn, P. J Digit Imaging (2018). https://doi.org/10.1007/s10278-018-0056-0
Acknowlegements
Code borrows heavily from pix2pix
评论