| import os | |
| from tqdm import tqdm | |
| import pickle | |
| import shutil | |
| speakers = ['seth', 'oliver', 'conan', 'chemistry'] | |
| source_data_root = "../expressive_body-V0.7" | |
| data_root = "D:/Downloads/SHOW_dataset_v1.0/ExpressiveWholeBodyDatasetReleaseV1.0" | |
| f_read = open('split_more_than_2s.pkl', 'rb') | |
| f_save = open('none.pkl', 'wb') | |
| data_split = pickle.load(f_read) | |
| none_split = [] | |
| train = val = test = 0 | |
| for speaker_name in speakers: | |
| speaker_root = os.path.join(data_root, speaker_name) | |
| videos = [v for v in data_split[speaker_name]] | |
| for vid in tqdm(videos, desc="Processing training data of {}......".format(speaker_name)): | |
| for split in data_split[speaker_name][vid]: | |
| for seq in data_split[speaker_name][vid][split]: | |
| seq = seq.replace('\\', '/') | |
| old_file_path = os.path.join(data_root, speaker_name, vid, seq.split('/')[-1]) | |
| old_file_path = old_file_path.replace('\\', '/') | |
| new_file_path = seq.replace(source_data_root.split('/')[-1], data_root.split('/')[-1]) | |
| try: | |
| shutil.move(old_file_path, new_file_path) | |
| if split == 'train': | |
| train = train + 1 | |
| elif split == 'test': | |
| test = test + 1 | |
| elif split == 'val': | |
| val = val + 1 | |
| except FileNotFoundError: | |
| none_split.append(old_file_path) | |
| print(f"The file {old_file_path} does not exists.") | |
| except shutil.Error: | |
| none_split.append(old_file_path) | |
| print(f"The file {old_file_path} does not exists.") | |
| print(none_split.__len__()) | |
| pickle.dump(none_split, f_save) | |
| f_save.close() | |
| print(train, val, test) | |