Diff of /src/reorganize.py [000000] .. [602ab8]

Switch to side-by-side view

--- 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"])