root_folder_train = '/content/gdrive/My Drive/Deep Learning FYP/Code/train/' root_folder_test = '/content/gdrive/My Drive/Deep Learning FYP/Code/test/' emo_folder = os.listdir(root_folder_train) # emo_folder包含 7 个训练图像文件夹 emo_folder_test = os.listdir(root_folder_test) # emo_folder 包含 7 个测试图像文件夹
def load_images_from_folder(root_folder,emo_folder,all_images,all_labels):
all_images = []
all_labels = []
for cls, fol in enumerate(emo_folder):
print(cls,fol)
all_images_per_emo = os.listdir(os.path.join(root_folder, fol))
for im in all_images_per_emo[:10]:
img = cv2.imread(os.path.join(root_folder, fol, im),cv2.IMREAD_GRAYSCALE)
all_images.append(img)
all_labels.append(cls)
return all_images
return all_labels
root_folder = 'directory/folderpath'
train_images = []
train_labels = []
train_data =load_images_from_folder(root_folder_train,emo_folders_train,train_images ,train_labels)