Spaces:
Sleeping
Sleeping
File size: 1,810 Bytes
99afdfe |
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 42 43 44 45 46 47 48 49 50 51 52 |
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)
|