Upload layer_diff_dataset/video/convert_json.py with huggingface_hub

layer_diff_dataset/video/convert_json.py

@@ -0,0 +1,17 @@
+import json
+
+data = json.load(open("/home/qianwen.qian/mnt_pai/sihui.jsh/layer_diff_dataset/video/video_test.json", "r"))
+json_path = '/home/qianwen.qian/mnt_pai/sihui.jsh/data/video_dataset/YoutubeVOS/train/meta_v2_test2.5_cog_base.json'
+meta_dict = json.load(open(json_path))["videos"]
+meta_info = list(meta_dict.values())
+# print(meta_dict[0])
+# 003234408d
+# exit(0)
+new_data = []
+# print(data)
+for i, vid_info in enumerate(data):
+    vid_info["name"] = meta_info[i]["fg_prompt"]+', '+meta_info[i]["bl_prompt"]
+    vid_info["bg_prompt"] = meta_info[i]["bg_prompt"]
+    new_data.append(vid_info)
+with open("/home/qianwen.qian/mnt_pai/sihui.jsh/layer_diff_dataset/video/video_test_new.json", "w") as f:
+    json.dump(new_data, f, indent=4)