File size: 546 Bytes
45950ff | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | # 扫出data path
import os
data_root_list = [
# 'data/2_gmr_retarget_rl_0_120',
# 'data/2_gmr_retarget_rl_200_300',
# "data/2_gmr_retarget_rl_400_500"
"/mnt/shenzhen2cephfs/capybarali/codes/neobot/data/motionmillion/2_gmr_retarget_full"
]
save_path = 'data/gmr_path.txt'
with open(save_path, 'w') as f:
for data_root in data_root_list:
for root, dirs, files in os.walk(data_root):
for file in files:
if file.endswith('.npz'):
f.write(os.path.join(root, file) + '\n') |