Update make_json.py

EscopeFile/make_json.py

+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)