网上找了代码,发现有点毛病,于是改了改,人家用pillow我换成opencv了,生成在脚本目录下面别找错位置了 # encoding:utf-8 import pandas as pd import numpy as np import cv2 import os emotions = { '0': 'anger', # 生气 '1': 'disgust', # 厌恶 '2': 'fear', # 恐惧 '3': 'happy', # 开心 '4': 'sad', # 伤心 '5': 'surprised', # 惊讶 '6': 'normal', # 中性 } # 创建文件夹 def createDir(dir): if os.path.exists(dir) is False: os.makedirs(dir) def saveImageFromFer2013(file): # 读取csv文件 faces_data = pd.read_csv(file) imageCount = 0 # 遍历csv文件内容,并将图片数据按分类保存 for index in range(len(faces_data)): # 解析每一行csv文件内容 emotion_data = faces_data.loc[index][0] image_data = faces_data.loc[index][1] usage_data = faces_data.loc[index][2] # 将图片数据转换成48*48 data_array = list(map(float, image_data.split())) data_array = np.asarray(data_array) image = data_array.reshape(48, 48) print(type(image)) # 选择分类,并创建文件名 dirName = usage_data emotionName = emotions[str(emotion_data)] # 图片要保存的文件夹 imagePath = os.path.join(dirName, emotionName) # 创建“用途文件夹”和“表情”文件夹 createDir(dirName) createDir(imagePath) # 图片文件名 imageName = os.path.join(imagePath, str(index) + '.jpg') cv2.imwrite(imageName, image) imageCount = index print('总共有' + str(imageCount) + '张图片') if __name__ == '__main__': saveImageFromFer2013('D:/fer2013.csv')
[数据处理][原创]fer2013数据集csv转jpg
最新推荐文章于 2024-03-21 12:54:59 发布