Upload layer_diff_dataset/make_json1.py with huggingface_hub

layer_diff_dataset/make_json1.py

@@ -0,0 +1,27 @@
+import json
+import os
+# 构造包含Transparant_dragdata_550各图片的images_rgba_path的JSON文件
+
+folder_path = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/Transparant_dragdata_550/im'
+# folder_path_1 = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/train/im'
+
+file_list = os.listdir(folder_path)
+file_list = [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 = []
+# 修改每个字典的"images"和"images_rgba"字段
+for name in file_list:
+    item = {}
+    item['images_rgba'] = os.path.join(folder_path,name)
+    data.append(item)
+# print(data_try)
+# 写入新的JSON文件
+with open('/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/Transparant_dragdata_550/im_rgba.json', 'w') as file:
+    json.dump(data, file, indent=4)
+
+print("New JSON file has been created successfully.")