This repo is used to prepare metal artifact dataset in DuDoNet and DuDoNet++. The clean and metal contaminated CT images are illustrated below, respectively. Note the red pixels indicate the metallic implants.
-
Create folders
database_MAR/images_train
anddatabase_MAR/images_val
. -
We put a sample image
images/000001_01_01
to run the domeprepare_image_database.m
. -
To simulate the whole MAR database. Download CT images according to step 3. Uncomment
line 120
and commentling 121
inprepare_image_database.m
. Modetrain\val
can be selected inline 32
. Note:mode = 'train'
will take more than 5 hours and require ~560 G hard disk storage. -
Download and unzip DeepLesion dataset to folder
images
. Images_png_01.zip~Images_png_10.zip is sufficient. The structure of the unzipped files should look like:images/ 000001_01_01/ 000001_02_01/ ...
Thanks Dr. Yanbo Zhang and Dr. Hengyong Yu for providing the metal artifact synthesis code.