File size: 1,139 Bytes
3bc8d9b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | import os
from batchgenerators.utilities.file_and_folder_operations import join, load_json, isfile, save_json, maybe_mkdir_p
import csv
with open('/storage/chenqi/X_data/annotatedtumor_txt/colon_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]
with open('/storage/chenqi/code/DiffTumor/STEP3.SegmentationModel/cross_eval/colon_tumor_data_fold/real_tumor_val_1.txt', 'r') as f:
val_liver=f.readlines()
val_liver = [os.path.basename(i.split(' ')[0]).split('.')[0] for i in val_liver]
cases_info = [*csv.DictReader(open('final_mapping_qichen.csv'))]
val_bdmap=[]
for i in cases_info:
if i['source_dataset'] == 'MSD-Colon':
for j in val_liver:
if j == i['source_id']:
val_bdmap.append(i['AbdomenAtlas_id'])
breakpoint()
write_content = []
train_liver = list(set(all_files) - set(val_bdmap))
write_content.append({"train":train_liver, "val":val_bdmap})
# breakpoint()
splits_file = 'splits_final_colon_bdmap.json'
save_json(write_content, splits_file)
|