卷积神经网络实现人脸表情识别

一、实现过程

1.1 下载数据集
https://github.com/truongnmt/smile-detection
在这里插入图片描述
1.2 根据猫狗数据集训练的方法来训练笑脸数据集

#coding=gbk
import os
import sys
def rename():
    path=input("请输入路径(例如D:\\\\picture):")
    name=input("请输入开头名:")
    startNumber=input("请输入开始数:")
    fileType=input("请输入后缀名(如 .jpg、.txt等等):")
    print("正在生成以"+name+startNumber+fileType+"迭代的文件名")
    count=0
    filelist=os.listdir(path)
    for files in filelist:
        Olddir=os.path.join(path,files)
        if os.path.isdir(Olddir):
            continue
        Newdir=os.path.join(path,name+str(count+int(startNumber))+fileType)
        os.rename(Olddir,Newdir)
        count+=1
    print("一共修改了"+str(count)+"个文件")

rename() 

在这里插入图片描述
重新运行一遍,把0改为1,unsmile改为smile
在这里插入图片描述

1.2 图片分类

import os, shutil #复制文件
# 原始目录所在的路径
# 数据集未压缩
original_dataset_dir1 = 'D:\\database\\smile-detection-master\\smile-detection-master\\datasets\\train_folder\\1'  ##笑脸
original_dataset_dir0 = 'D:\\database\\smile-detection-master\\smile-detection-master\\datasets\\train_folder\\0'  ##非笑脸
# 我们将在其中的目录存储较小的数据集
base_dir = 'D:\\database\\smile-detection-master\\smile-detection-master1'
os.mkdir(base_dir)

# # 训练、验证、测试数据集的目录
train_dir = os.path.join(base_dir, 'train')
os.mkdir(train_dir)
validation_dir = os.path.join(base_dir, 'validation')
os.mkdir(validation_dir)
test_dir = os.path.join(base_dir, 'test')
os.mkdir(test_dir)

# 猫训练图片所在目录
train_cats_dir = os.path.join(train_dir, 'smile')
os.mkdir(train_cats_dir)

# 狗训练图片所在目录
train_dogs_dir = os.path.join(train_dir, 'unsmile')
os.mkdir(train_dogs_dir)

# 猫验证图片所在目录
validation_cats_dir = os.path.join(validation_dir, 'smile')
os.mkdir(validation_cats_dir)

# 狗验证数据集所在目录
validation_dogs_dir = os.path.join(validation_dir, 'unsmile')
os.mkdir(validation_dogs_dir)

# 猫测试数据集所在目录
test_cats_dir = os.path.join(test_dir, 'smile')
os.mkdir(test_cats_dir)

# 狗测试数据集所在目录
test_dogs_dir = os.path.join(test_dir, 'unsmile')
os.mkdir(test_dogs_dir)

# 将前1000张笑脸图像复制到train_cats_dir
fnames = ['smile.{}.jpg'.format(i) for i in range(1000)]
for fname in fnames:
    src = os.path.join(original_dataset_dir1, fname)
    dst = os.path.join(train_cats_dir, fname)
    shutil.copyfile(src, dst)

# 将下500张笑脸图像复制到validation_cats_dir
fnames = ['smile.{}.jpg'.format(i) for i in range(500
  • 0
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
好的,以下是一个简单的卷积神经网络实现人脸表情识别检测的代码,包含了图片实际检测的部分: 首先,我们需要导入一些必要的库: ```python import numpy as np import cv2 import tensorflow as tf from tensorflow import keras from tensorflow.keras import layers, models ``` 然后,我们需要准备数据。我们将使用FER2013数据集,该数据集包含了7种不同的人脸表情类别。我们将使用以下代码加载和预处理数据: ```python # Load data with open("fer2013.csv") as f: content = f.readlines() lines = np.array(content) num_of_instances = lines.size print("Number of instances: ",num_of_instances) print("Instance length: ",len(lines[1].split(",")[1].split(" "))) # Initialize train and test set x_train, y_train, x_test, y_test = [], [], [], [] # Transfer train and test set data for i in range(1,num_of_instances): try: emotion, img, usage = lines[i].split(",") val = img.split(" ") pixels = np.array(val, 'float32') emotion = keras.utils.to_categorical(emotion, 7) if 'Training' in usage: y_train.append(emotion) x_train.append(pixels) elif 'PublicTest' in usage: y_test.append(emotion) x_test.append(pixels) except: print("", end="") # Data transformation x_train = np.array(x_train, 'float32') y_train = np.array(y_train, 'float32') x_test = np.array(x_test, 'float32') y_test = np.array(y_test, 'float32') # Normalization x_train /= 255 x_test /= 255 # Reshape x_train = x_train.reshape(x_train.shape[0], 48, 48, 1) x_test = x_test.reshape(x_test.shape[0], 48, 48, 1) ``` 接下来,我们将定义一个简单的卷积神经网络模型: ```python model = models.Sequential() # 1st convolution layer model.add(layers.Conv2D(64, (3, 3), activation='relu', input_shape=(48,48,1))) model.add(layers.MaxPooling2D(pool_size=(2, 2))) model.add(layers.Dropout(0.25)) # 2nd convolution layer model.add(layers.Conv2D(128, (5, 5), activation='relu')) model.add(layers.MaxPooling2D(pool_size=(2, 2))) model.add(layers.Dropout(0.25)) # 3rd convolution layer model.add(layers.Conv2D(512, (3, 3), activation='relu')) model.add(layers.MaxPooling2D(pool_size=(2, 2))) model.add(layers.Dropout(0.25)) # 4th convolution layer model.add(layers.Conv2D(512, (3, 3), activation='relu')) model.add(layers.MaxPooling2D(pool_size=(2, 2))) model.add(layers.Dropout(0.25)) # Flatten layer model.add(layers.Flatten()) # Fully connected layer 1st layer model.add(layers.Dense(256, activation='relu')) model.add(layers.Dropout(0.25)) # Fully connected layer 2nd layer model.add(layers.Dense(512, activation='relu')) model.add(layers.Dropout(0.25)) model.add(layers.Dense(7, activation='softmax')) ``` 接下来,我们将编译模型并进行训练: ```python model.compile(loss='categorical_crossentropy', optimizer=keras.optimizers.Adam(lr=0.0001, decay=1e-6), metrics=['accuracy']) model_info = model.fit( x_train, y_train, batch_size=64, epochs=50, validation_data=(x_test, y_test), shuffle=True ) ``` 最后,我们将使用OpenCV读取一张人脸图片并进行表情识别: ```python # Load the cascade face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml') # Read the input image img = cv2.imread('test.jpg') # Convert into grayscale gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) # Detect faces faces = face_cascade.detectMultiScale(gray, 1.1, 4) # Draw rectangle around the faces and predict emotion for (x, y, w, h) in faces: roi_gray = gray[y:y + h, x:x + w] roi_gray = cv2.resize(roi_gray, (48, 48)) img_pixels = tf.keras.preprocessing.image.img_to_array(roi_gray) img_pixels = np.expand_dims(img_pixels, axis=0) img_pixels /= 255 predictions = model.predict(img_pixels) # Find max indexed array max_index = np.argmax(predictions[0]) emotions = ('angry', 'disgust', 'fear', 'happy', 'sad', 'surprise', 'neutral') predicted_emotion = emotions[max_index] # Draw rectangle around the face cv2.rectangle(img, (x, y), (x + w, y + h), (255, 0, 0), 2) # Draw label cv2.putText(img, predicted_emotion, (x, y), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2) # Display the output cv2.imshow('img', img) cv2.waitKey() ``` 希望这个代码能够帮到你!

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值