Upload layer_diff_dataset/copy_gt.py with huggingface_hub

layer_diff_dataset/copy_gt.py

@@ -0,0 +1,13 @@
+import os
+import shutil
+from tqdm import tqdm
+
+image_dir_path = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/try/gt_inp_iic_dilate'
+image_dir_path_1 = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/train/gt_inp_iic_gt_dilate'
+image_dir_path_2 = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/try/gt_inp_iic_gt_dilate'
+os.makedirs(image_dir_path_2,exist_ok=True)
+image_dir = os.listdir(image_dir_path)
+image_dir.sort()
+pbar = tqdm(enumerate(image_dir),total=len(image_dir))
+for idx, image_name in pbar:
+    shutil.copy(os.path.join(image_dir_path_1,image_name),os.path.join(image_dir_path_2,image_name))