drop_modality.py

## This script create a pseudo validation set during the validation stage and a test set during the final evaluation. 

import os
import random 
import numpy as np
import shutil

# the target validation set foler and the new folder, please replace it with yours
val_set_folder = 'ASNR-MICCAI-BraTS2023-GLI-Challenge-TrainingData' 
val_set_missing = 'pseudo_val_set'

if not os.path.exists(val_set_missing):
    os.mkdir(val_set_missing)

# create a pseudo validation set by randomly dropping one modality
np.random.seed(123456)  # fix random seed 
modality_list = ['t1c', 't1n', 't2f', 't2w']  # the list of modalities in the given folder
folder_list = os.listdir(val_set_folder)
folder_list.sort()
drop_index = np.random.randint(0, 4, size=len(folder_list))


for count, ff in enumerate(folder_list):
    if not os.path.exists(os.path.join(val_set_missing, ff)):
        os.mkdir(os.path.join(val_set_missing, ff))
    
    file_list = os.listdir(os.path.join(val_set_folder, ff))

    for mm in file_list:
        print(modality_list[drop_index[count]] + ' is droppd for case ' + mm)
        if not modality_list[drop_index[count]] in mm:
            shutil.copyfile(os.path.join(val_set_folder, ff, mm), os.path.join(val_set_missing, ff, mm))