23 lines
806 B
Python
23 lines
806 B
Python
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) |