分割RAF_DB数据集
'''
1 Surprise
2 Fear
3 Disgust
4 Happiness
5 Sadness
6 Anger
7 Neutral
'''
import os
import shutil
typelist = ['Surprise','Fear','Disgust','Happiness','Sadness','Anger','Neutral']
label_list_txt_path = '.\\list_patition_label.txt'
input_path = '.\\inputdataset\\aligned'
output_path = '.\\outputdataset'
with open(label_list_txt_path, 'r') as f:
informations = f.readlines()
number = 1
for i in informations:
model_mode = i.split(" ")[0].split("_")[0]
index = eval(i.split(" ")[-1])
old_name = os.path.join(input_path,str(i.split(" ")[0]))
if model_mode == 'train':
new_name = os.path.join(output_path,str(model_mode)+"\\"+str(typelist[index-1])+"\\"+str(i.split(" ")[0]))
if model_mode == 'test':
new_name = os.path.join(output_path,str(model_mode)+"\\"+str(typelist[index-1])+"\\"+str(i.split(" ")[0]))
shutil.copy(old_name,new_name)
print("处理第%d张..."%number)
number = number + 1
print("处理结束...")