File size: 1,256 Bytes
eb71a72 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import numpy as np
import os
import sys
music_dir = "data/finedance/music_npy"
motion_dir = "data/finedance/motion_fea319"
music_out = "data/finedance/div_by_time/music_npy_"
motion_out = "data/finedance/div_by_time/motion_fea319_"
timelen = 120
music_out = music_out + str(timelen)
motion_out = motion_out + str(timelen)
if not os.path.exists(music_out):
os.makedirs(music_out)
if not os.path.exists(motion_out):
os.makedirs(motion_out)
for file in os.listdir(motion_dir):
if file[-3:] != 'npy':
print(file[-3:])
continue
name = file.split(".")[0]
music_fea = np.load(os.path.join(music_dir, file))
motion_fea = np.load(os.path.join(motion_dir, file))
max_length = min(music_fea.shape[0], motion_fea.shape[0])
iters = (max_length//timelen)
max_length = iters*timelen
music_fea = music_fea[:max_length, :]
motion_fea = motion_fea[:max_length, :]
for i in range(iters):
music_clip = music_fea[i*timelen: (i+1)*timelen, :]
motion_clip = motion_fea[i*timelen: (i+1)*timelen, :]
np.save(os.path.join(music_out, name + "z@" + str(i).zfill(3) + ".npy"), music_clip)
np.save(os.path.join(motion_out, name + "z@" + str(i).zfill(3) + ".npy"), motion_clip)
|