人脸表情识别数据集FER2013下载与处理指南

最新推荐文章于 2024-10-29 19:27:12 发布

徐梁珩Walton

最新推荐文章于 2024-10-29 19:27:12 发布

阅读量1.4k

点赞数 5

本文链接：https://blog.csdn.net/gitblog_06524/article/details/142398558

版权

人脸表情识别数据集FER2013下载与处理指南

人脸表情识别1下载fer2013数据集和数据的处理项目地址: https://gitcode.com/Resource-Bundle-Collection/8ffa9

简介

本资源文件提供了FER2013数据集的下载链接以及数据处理方法。FER2013数据集是一个广泛用于人脸表情识别任务的数据集，包含了多种情绪类别的图像数据。通过本指南，您可以轻松获取并处理该数据集，为后续的模型训练和实验做好准备。

数据集概述

FER2013数据集包含了35886张人脸表情图片，分为训练集、验证集和测试集。每张图片的大小为48x48像素，灰度图像。数据集中的表情类别包括：

0: 生气
1: 厌恶
2: 恐惧
3: 开心
4: 伤心
5: 惊讶
6: 中性

数据集下载

您可以通过提供的链接下载FER2013数据集。下载完成后，您将获得一个名为fer2013.csv的文件。

数据处理步骤

分割数据集：将fer2013.csv文件中的数据分割为训练集、验证集和测试集。
转换为图片格式：将分割后的数据转换为图片格式，并按照类别分别存储在不同的文件夹中。

代码示例

以下是数据处理的部分代码示例，帮助您将CSV文件转换为图片格式：

import csv
import os
from PIL import Image
import numpy as np

datasets_path = r'path_to_datasets'
train_csv = os.path.join(datasets_path, 'train.csv')
val_csv = os.path.join(datasets_path, 'val.csv')
test_csv = os.path.join(datasets_path, 'test.csv')

train_set = os.path.join(datasets_path, 'train')
val_set = os.path.join(datasets_path, 'val')
test_set = os.path.join(datasets_path, 'test')

for save_path, csv_file in [(train_set, train_csv), (val_set, val_csv), (test_set, test_csv)]:
    if not os.path.exists(save_path):
        os.makedirs(save_path)
    num = 1
    with open(csv_file) as f:
        csvr = csv.reader(f)
        header = next(csvr)
        for i, (label, pixel) in enumerate(csvr):
            pixel = np.asarray([float(p) for p in pixel.split()]).reshape(48, 48)
            subfolder = os.path.join(save_path, label)
            if not os.path.exists(subfolder):
                os.makedirs(subfolder)
            im = Image.fromarray(pixel).convert('L')
            image_name = os.path.join(subfolder, '{:05d}.jpg'.format(i))
            im.save(image_name)