diff --git a/convert_to_uv_local_single.py b/convert_to_uv_local_single.py index 5181cbf..d023a4b 100644 --- a/convert_to_uv_local_single.py +++ b/convert_to_uv_local_single.py @@ -530,9 +530,9 @@ if __name__ == '__main__': # for i in range(uv_gt.shape[0]): # np.savez_compressed(os.path.join(folder, id, 'gt_uv', garment + '_' + str(i) + '.npz'), uv_gt=uv_gt[i, :, :, :]) - os.makedirs(os.path.join(folder, id, f'curvature_uv_{neighbourhood}'), exist_ok=True) + os.makedirs(os.path.join(folder, f'uv_curvature_{neighbourhood}'), exist_ok=True) for i in range(uv_curvature.shape[0]): - np.savez_compressed(os.path.join(folder, id, f'curvature_uv_{neighbourhood}', id + '_' + str(i) + '.npz'), curvature_uv=uv_curvature[i, :, :]) + np.savez_compressed(os.path.join(folder, f'uv_curvature_{neighbourhood}', id + '_' + str(i) + '.npz'), curvature_uv=uv_curvature[i, :, :]) # os.makedirs(os.path.join(folder, id, 'uv_unposed'), exist_ok=True) # for i in range(uv_unposed.shape[0]): diff --git a/fix_curv_uv_loc.py b/fix_curv_uv_loc.py new file mode 100644 index 0000000..b8b7af6 --- /dev/null +++ b/fix_curv_uv_loc.py @@ -0,0 +1,23 @@ +import os +import shutil +import tqdm + +if __name__ == '__main__': + folder = '/home/workspace/data/dataset/local-cache/train_skirt' + is_skirt = True + skirt_modifier = '_skirt' if is_skirt else '' + frames_file = 'frames' + skirt_modifier + '_10p.txt' + neighbourhood = 10 + + frames = [] + with open(os.path.join(folder, frames_file), 'r') as f: + lines = f.readlines() + lines = [line.strip() for line in lines] + frames = lines + + os.makedirs(os.path.join(folder, f'uv_curvature_{neighbourhood}'), exist_ok=True) + for frame in tqdm.tqdm(frames): + id = frame.split('_')[0] + src = os.path.join(folder, f'{id}', 'curvature_uv_10', f'{frame}.npz') + dst = os.path.join(folder, f'uv_curvature_{neighbourhood}', f'{frame}.npz') + shutil.move(src, dst) \ No newline at end of file