2021-02-17

csv文件转图片(fer2013)
1、代码

encoding:utf-8

import pandas as pd
import numpy as np

noinspection PyUnresolvedReferences

import scipy.misc as sm
import os

noinspection PyUnresolvedReferences

from PIL import Image

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,low_memory=False)
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)

    # 选择分类，并创建文件名
    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')
    #sm.toimage(image).save(imageName)
    Image.fromarray(image).convert('L').save(imageName)
    imageCount = index
print('总共有' + str(imageCount) + '张图片')

if name == ‘main’:
saveImageFromFer2013(‘fer2013.csv’)