--- a +++ b/src/reorganize.py @@ -0,0 +1,34 @@ +import os +import glob +import shutil +from tqdm import * +from subprocess import check_call + + +def create_dir(path): + if not os.path.isdir(path): + os.makedirs(path) + return + + +src_dir_list = ["ADNI1_Screening_AD", "ADNI1_Screening_NC", + "ADNI2_Screening_AD", "ADNI2_Screening_NC"] +dst_dir_list = ["AD", "NC", "AD", "NC"] + +parent_dir = os.path.dirname(os.getcwd()) +data_dir = os.path.join(parent_dir, "data") + +for in_dir, out_dir in zip(src_dir_list, dst_dir_list): + data_src_dir = os.path.join(data_dir, in_dir) + data_dst_dir = os.path.join(data_dir, "ADNI", out_dir) + create_dir(data_dst_dir) + + print("Move files\nfrom: {0}\nto {1}".format(data_src_dir, data_dst_dir)) + + subjects = os.listdir(data_src_dir) + for subject in tqdm(subjects): + subj_dir = os.path.join(data_src_dir, subject) + src_path = glob.glob(os.path.join(subj_dir, "*", "*", "*", "*"))[0] + dst_path = os.path.join(data_dst_dir, subject + ".nii") + shutil.copyfile(src_path, dst_path) + check_call(["gzip", dst_path, "-f"])