# 扫出data path import os data_root = "data/smplx_data" save_path = 'data/final_data/all.txt' cnt = 0 # with open(save_path, 'w') as f: for root, dirs, files in os.walk(data_root): for file in files: if file.endswith('.npy'): cnt += 1 print(cnt) # f.write(os.path.join(root, file) + '\n')