--- a +++ b/Preprocessing Medical Data Pipeline/createDirectories.py @@ -0,0 +1,46 @@ +import os +import shutil + +def createDirectoriesfunc(): + def make_if_dont_exist(folder_path,overwrite): + """ + creates a folder if it does not exists + input: + folder_path : relative path of the folder which needs to be created + over_write :(default: False) if True overwrite the existing folder + """ + if os.path.exists(folder_path): + + if not overwrite: + print(f"{folder_path} exists.") + else: + print(f"{folder_path} overwritten") + shutil.rmtree(folder_path) + os.makedirs(folder_path) + + else: + os.makedirs(folder_path) + print(f"{folder_path} created!") + + + + base_directory = 'D:/MRI - Tairawhiti (User POV)' + new_folder_name = ['nnUNet Data', 'Raw NIFITI Segmentation Masks (3D Slicer Output)', 'Pre-Trained Models (Google Colab)', 'Model Code', 'Raw DICOM MRI Scans', 'Patient Segmentation Tasks (Google Colab)'] + overwrite = False + for new_fname in new_folder_name: + folder_path = os.path.join(base_directory, new_fname) + make_if_dont_exist(folder_path, overwrite) + + + base_directory = 'D:/MRI - Tairawhiti (User POV)/nnUNet Data' + new_folder_name = ['masks', 'multiclass_masks', 'scans'] + for new_fname in new_folder_name: + folder_path = os.path.join(base_directory, new_fname) + make_if_dont_exist(folder_path, overwrite) + + + base_directory = 'D:/MRI - Tairawhiti (User POV)/nnUNet Data/masks' + new_folder_name = ['FEMUR', 'FIBULA', 'TIBIA', 'PELVIS'] + for new_fname in new_folder_name: + folder_path = os.path.join(base_directory, new_fname) + make_if_dont_exist(folder_path, overwrite)