diff --git a/merge_folders.py b/merge_folders.py new file mode 100644 index 0000000..9084050 --- /dev/null +++ b/merge_folders.py @@ -0,0 +1,32 @@ +import os +import tqdm +import shutil + + + +if __name__ == '__main__': + + folder_in = '/home/workspace/data/dataset/train_skirt_25' + folder_out = '/home/workspace/data/dataset/train_skirt_25_2' + + ids = os.listdir(folder_in) + + for id in tqdm.tqdm(ids): + garment_dir = os.path.join(folder_in, id) + + if not os.path.isdir(garment_dir): + continue + + garments = [f.replace('.pc16','') for f in os.listdir(garment_dir) if f.find('_') == -1 and f != 'info.mat' and not f.endswith('.obj')] + # assert len (garments) == 1 + if len (garments) != 1: + continue + garment = garments[0] + + # shutil.copytree(os.path.join(folder_in, id, 'mesh_unposed'), os.path.join(folder_out, id, 'mesh_unposed'), dirs_exist_ok=True) + # shutil.copytree(os.path.join(folder_in, id, 'mesh_unposed_no_shape'), os.path.join(folder_out, id, 'mesh_unposed_no_shape'), dirs_exist_ok=True) + # shutil.copytree(os.path.join(folder_in, id, 'mesh_posed_body_no_orient'), os.path.join(folder_out, id, 'mesh_posed_body_no_orient'), dirs_exist_ok=True) + shutil.copytree(os.path.join(folder_in, id, 'body_posed_uv'), os.path.join(folder_out, id, 'body_posed_uv'), dirs_exist_ok=True) + shutil.copytree(os.path.join(folder_in, id, 'uv_posed_body_no_orient'), os.path.join(folder_out, id, 'uv_posed_body_no_orient'), dirs_exist_ok=True) + # shutil.copy(os.path.join(folder_in, id, 'curvature_25_raw.npy'), os.path.join(folder_out, id, 'curvature_25_raw.npy')) + shutil.copy(os.path.join(folder_in, id, 'body_static_uv.npy'), os.path.join(folder_out, id, 'body_static_uv.npy'))