Upload layer_diff_dataset/make_json copy 2.py with huggingface_hub

layer_diff_dataset/make_json copy 2.py

@@ -0,0 +1,31 @@
+import json
+import os
+# 修改原train json的prompt，增加类别
+
+# folder_path = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/try/im'
+# folder_path_1 = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/train/im'
+
+# file_list = os.listdir(folder_path)
+# file_list = [os.path.join(folder_path_1,i) for i in file_list if i.endswith('.jpg')]
+# file_list.sort()
+# print(file_list)
+# 读取原始JSON文件
+with open('/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/train/im_rgba.json', 'r') as file:
+    data = json.load(file)
+
+# data_try = []
+# # 修改每个字典的"images"和"images_rgba"字段
+# for item in data:
+#     if item["images"] not in file_list:
+#         data_try.append(item)
+# print(data_try)
+for item in data:
+    class_name = item["images"].split('#')[3]
+    item['prompt'] = class_name + ',' + item['prompt']
+    # print(item)
+    # break
+# 写入新的JSON文件
+with open('/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/train/im_rgba_class.json', 'w') as file:
+    json.dump(data, file, indent=4)
+
+print("New JSON file has been created successfully.")