import os
from batchgenerators.utilities.file_and_folder_operations import join, load_json, isfile, save_json, maybe_mkdir_p
import csv


all_files = os.listdir('/storage/chenqi/X_data/esophagus_BodyMapPro')
val_files = load_json('/storage/chenqi/code/GEM-3D-ct/data_split/splits_final_eso_bdmap.json')[0]['val']


write_content = []
train_liver = list(set(all_files) - set(val_files))
# breakpoint()
write_content.append({"train":train_liver, "val":val_files})

# breakpoint()

splits_file = 'splits_eso.json'
save_json(write_content, splits_file)