提取文件夹的图片并将其合成为数据集(fer2013)

最新推荐文章于 2022-06-21 19:44:10 发布

qq_35482604

最新推荐文章于 2022-06-21 19:44:10 发布

阅读量693

点赞数 2

本文链接：https://blog.csdn.net/qq_35482604/article/details/104925668

版权

在这里插入图片描述
链接：https://pan.baidu.com/s/1k3FfWMd_aSyISCtxj9nvDQ
提取码：8x2j

# -*- coding: UTF-8 -*-
#@Time : 2020/3/17 @Author : SUNLIN
import os
import cv2
import numpy as np

file='../data/fer2013/Training'
emotions=['anger','disgust','fear','happy','neutral','sad','surprised']
train_x=[]
for i in range(len(emotions)):
    folder=os.path.join(file,emotions[i])
    images_name=os.listdir(folder)
    for j in range(len(images_name)):
        image=cv2.imread(os.path.join(folder,images_name[j]),0)
        image=image[:,:,np.newaxis]
        train_x.append(image)
train_x=np.array(train_x).astype('float32')/255
print(train_x.shape)