| import os | |
| from batchgenerators.utilities.file_and_folder_operations import join, load_json, isfile, save_json, maybe_mkdir_p | |
| import csv | |
| train_files = load_json('/storage/chenqi/code/GEM-3D-ct/data_split/splits_final_uterus.json')[0]['train'] | |
| val_files = load_json('/storage/chenqi/code/GEM-3D-ct/data_split/splits_final_uterus.json')[0]['val'] | |
| with open('/storage/chenqi/X_data/annotatedtumor_txt/endometrioma_tumor_cases.txt', 'r') as f: | |
| all_files=f.readlines() | |
| all_files = [os.path.basename(i.split(' ')[0]).split('.')[0] for i in all_files] | |
| all_files = [i.split('\n')[0] for i in all_files] | |
| print(val_files) | |
| val_files = ['BDMAP_'+str(int(i.split('BDMAP_endometrioma_')[1])+22411).zfill(8) for i in val_files] | |
| 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_uterus.json' | |
| save_json(write_content, splits_file) | |