diff --git a/EscopeFile/make_json.py b/EscopeFile/make_json.py index 8b137891791fe96927ad78e64b0aad7bded08bdc..18bbb272d5be63d89796f7da77cd7803410ea5d2 100644 --- a/EscopeFile/make_json.py +++ b/EscopeFile/make_json.py @@ -1 +1,35 @@ +import json +import os +image_path = r"F:\data\cta-data\coronary-144\train\images" +label_path = r"F:\data\cta-data\new_aorta144\labels\merge_144case" +image_files = [file for file in os.listdir(image_path)] +label_files = [file for file in os.listdir(label_path)] +list_data = [] +for i in range(len(image_files)): + image = image_files[i] + label = label_files[i] + print(image) + print(label) + dic = { + "image": "images/" + image, + "label": "labels/" + label + } + list_data.append(dic) +data = { + "name": "EscopeCT_Seg", + "modality": { + "0": "CT" + }, + "labels": { + "0": "background", + "1": "coronary", + "2": "calcification", + "3": "aorta" + }, + "training": list_data +} + +file_name = 'dataset.json' #通过扩展名指定文件存储的数据为json格式 +with open(file_name, 'w') as file_object: + json.dump(data, file_object)