1、摘要

本项目将在Android上实现一种通过识别表情类别,从而给人脸戴上不同样式脸谱的AR软件,效果如下:

基于表情识别的脸谱换脸AR安卓APP效果演示

通过深度学习和Keras训练一个人脸表情识别的卷积神经网络,然后使用TensorFlow Lite转换为tflite文件,部署到Android平台。

想要实现这样一个软件,核心就是两部分:
1)使用卷积神经网络训练一个人脸表情识别模型,
2)将训练好的模型移植到Android平台,同时在Android实现脸谱AR效果,并结合表情识别模型的识别结果,渲染不同的脸谱样式

本文讲第一部分,如何使用Keras训练一个人脸表情识别的卷积神经网络。

第二部分见:基于卷积神经网络的人脸表情识别应用–AR川剧变脸(二)

2、数据集处理

数据集我们使用FER2013PLUS人脸表情识别数据集,大概有35000张图片,每张图片是48*48分辨率的灰度图。

下图展示7个表情类别的数据集样本量,可以看出类别 angry、happy、sad、surprised、normal 这 5 种表情数量较多,而 disgust、fear这 2 种表情较少。

在实验中为了防止出现过拟合现象需要进行数据集扩充,提高模型的泛化能力。数据增强是指对输入的图像进行随机的图像处理,比如旋转、位移、光照强度、裁剪等,对于计算机而言,经过位移后的图像和原图像是两幅不同的图像,但具有相同的标签,从而扩充了已有的数据集。而通过使用大量不同的数据集来训练网络,进一步提高深度模型的泛化能力,使得网络即不过拟合也不欠拟合,达到最佳状态。

模型的实际使用常见是自由场景,自由场景中光照亮度的不同会影响模型的性能,添加随机光照,同时为了扩充数据集和对非正常角度人脸的识别增加随机旋转、随机聚焦和水平翻转。下图 展示对一张 48*48 尺寸,表情为 happy 的灰度图进行随机亮度、随机旋转、随机聚焦和随机水平翻转的效果。可以看出,小型数据集借此技术可以实现数十倍的规模扩充,而数据集规模的提升可以提升网络的泛化能力和鲁棒性,提高网络在场景下的识别准确率。

在Keras中,我们需要使用数据加载器加载数据,最大化硬件利用率,幸运的是Keras已经提供了方法,我们直接使用即可:

def getDenerator(fer2013plus_path="datasets/fer2013plus", batch_size=32):"""我们传递数据集的位置和batch_size,然后函数返回用于训练和测试的数据加载器,用于训练时投喂数据:param fer2013plus_path:fer2013plus数据集目录:param batch_size::return:"""train_datagen = ImageDataGenerator(    # 定义数据增强的方式brightness_range=(0.7, 1.3),  # 随机亮度featurewise_center=False,featurewise_std_normalization=False,rotation_range=10,  # 随机旋转zoom_range=(0.7, 1.3),  # 聚焦horizontal_flip=True)  # 水平翻转train_generator = train_datagen.flow_from_directory(fer2013plus_path+"/train",  # 告诉Keras图片的位置target_size=(48, 48),       # 将图片缩放到指定大小color_mode="grayscale",     # 因为是灰度图batch_size=batch_size,        # 一次性加载多少图片,越大训练越快,但是需要更多的显存class_mode='categorical')    test_datagen = ImageDataGenerator()  # 水平翻转test_generator = test_datagen.flow_from_directory(fer2013plus_path+"/test",target_size=(48, 48),color_mode="grayscale",batch_size=batch_size,class_mode='categorical')return train_generator,test_generator

3、模型设计

模型设计参考VGG网络,也就是Conv+Conv+Pool式的堆叠。

如下图是VGG的结构,可以看出就是Conv+Conv+Pool式的堆叠,最后使用全连接层,实现1000类别的分类。

但是我们人脸表情识别数据集比较小,仅有3万多张图片,模型太大容易过拟合,所以以VGGNet为蓝本,设计一个比较小的模型,如下图所示,也是conv+conv+pool式,但是更小。

因为我们的表情是7中分类,所以输出的全连接层长度是7。

在Keras中定义模型结构其实而很简单,代码如下:

def getVXSlim():model = Sequential()# block1model.add(Conv2D(64, (5, 5), input_shape=(48, 48, 1), activation='relu', padding='same'))model.add(Conv2D(64, (5, 5), activation='relu', padding='same'))model.add(BatchNormalization())model.add(MaxPooling2D(pool_size=(2, 2)))# block2model.add(Conv2D(128, (5, 5), activation='relu', padding='same'))model.add(Conv2D(128, (5, 5), activation='relu', padding='same'))model.add(BatchNormalization())model.add(MaxPooling2D(pool_size=(2, 2)))# block3model.add(Conv2D(256, (3, 3), activation='relu', padding='same'))model.add(Conv2D(256, (3, 3), activation='relu', padding='same'))model.add(BatchNormalization())model.add(MaxPooling2D(pool_size=(2, 2)))# single convmodel.add(Conv2D(64, (3, 3), padding='same'))# classifiermodel.add(Flatten())      # 将多维的数据展成一列,可以理解为将图像展开成一列model.add(Dense(7))            # 堆叠一个全连接层,长度为7model.add(BatchNormalization())model.add(Dense(7))         # 堆叠一个全连接层,长度为7# configmodel.add(Activation('softmax'))   # 最后使用softmax激活函数,将输出转换为概率model.compile(loss='categorical_crossentropy', metrics=['accuracy'], optimizer='adam')  # 定义优化器return model

在上面的代码中使用了Conv2D、BatchNormalization、MaxPooling2D、Dense:

  • Conv2D:提取特征,将图像分解为多个特征的组合
  • BatchNormalization:将数据的分布变换为正态分布,再映射到其他分布,可以加快模型的训练速度
  • MaxPooling2D:可以缩减图像的尺寸
  • Dense:全连接层,用于综合所有特征进行判断类别

4、训练

其实到这里,最重要的数据加载和模型定义就写好了,就可以开始训练了。训练了100个轮次,然后开始训练。

从图中可以看到,训练集精度和损失在 45epoch 左右进入缓慢优化阶段,在 70epoch左右趋于稳定。测试集精度和损失在 45epoch 左右趋于稳定,最终测试集精度达到 83%,损失低于 0.3,取得了较好的表情识别效果。

都说深度学习是黑盒,我们来看看训练出来的模型的可视化效果。

在可视化中间输出部分,通过卷积网络的每层都会输出结果,通过将结果可视化可以看到输入图片在网络中的流动形式,在这里展示第 3 层的 max_pooling2d、第 4 层的conv2d_2 和第 8 层的 conv2d_4 的输出结果。在图 中可以看出,在开始阶段实际上是各种边缘检测器的集合,在这一阶段,激活几乎保留了输入图像中所有的信息。在第二幅中,虽然仍有边缘检测的痕迹,但是此时更加像是对某些局部部位的特征检测,随着层数的加深,中间激活的结果变得越来越抽象,也更加难以直观地理解。在第三层中几乎无法和原图联系起来。所以抽象是卷积神经网络的一大能力,而这种能力也正是人类所拥有的。

我们将本文设计的模型和VGG9,以及mini_Xception进行对比:

从图中可以看出,本文的VXSlim网络的验证集精度最高,达到了 83%,VGG9 达到了 82%,mini_Xception 最低,只有79%

那么本文的模型在7种表情类别上的效果如何?可以看到最好的是开心类别,正确率达到了91%!

我们把每一个类的正确识别率用曲线表示,可以看到本文设计的模型在7种类别上的正确率都是最高的,很nice!

5、效果展示

通过摄像头进行表情识别效果展示

VXSlim演示_x264

代码资源:
链接:https://pan.baidu.com/s/1Y0xWDmz_lg04PTiukoBVgQ?pwd=ozn5
提取码:ozn5

基于卷积神经网络的人脸表情识别应用--AR川剧变脸(一)相关推荐

  1. 基于卷积神经网络的人脸表情识别应用--AR川剧变脸(二)

    本项目将在Android上实现一种通过识别表情类别,从而给人脸戴上不同样式脸谱的AR软件,效果如下: 基于表情识别的脸谱换脸AR安卓APP效果演示 想要实现这样一个软件,核心就是两部分: 1)使用卷积 ...

  2. 基于卷积神经网络的人脸表情识别(JAFFE篇)

    一.前言 人脸表情识别依然是计算机视觉中的研究重点,那就意味着它还有水文章的可能性.本专题将专门讨论各种表情识别的研究方法,当然我们从最简单的单一卷积网络开始.本文中的代码直接复制到电脑端(pytho ...

  3. 基于深度神经网络的人脸表情识别

    Preprocessing Fer2013 数据集 链接:fer2013.csv 提取码:mzvb 复制这段内容后打开百度网盘手机App,操作更方便哦 参考 下载fer2013数据集和数据的处理 基于 ...

  4. 基于卷积网络的人脸表情识别及其应用

    基于卷积网络的人脸表情识别及其应用 前言 一.人脸表情识别技术现状 二.卷积神经网络技术概述 1.卷积神经网络图像理解过程 2.卷积神经网络组成结构 3.卷积神经网络的优势 三.人脸表情识别卷积网络模 ...

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

    文章目录 一.实现过程 1.1 下载数据集 1.2 根据猫狗数据集训练的方法来训练笑脸数据集 1.2 图片分类 1.3 作为健全性检查,计算一下在每个训练分割中我们有多少图片(训练/验证/测试): 1 ...

  6. 基于卷积神经网络的辛普森角色识别

    摘 要 基于深度学习在图像分类领域的优异性能,本文研究基于图像识别技术的辛普森角色自动识别方法.首先采集18个角色的16503幅辛普森角色图像数据集,然后在CNN模型框架下,修改最顶端的全连接层与分类 ...

  7. 博士论文——基于卷积神经网络的人脸识别研究 __张燕红

    论文题目 作者 年份 关键词 链接 备注 基于卷积神经网络的人脸识别研究 张燕红 2018 人脸识别:卷积神经网络:特征提取:分块策略:正则化 博士论文 摘要:随着信息技术的蓬勃发展,人们的学习和生活 ...

  8. 基于卷积神经网络的人脸识别(自我拍摄获取数据集)

    基于卷积神经网络的人脸识别 完整代码.数据请见:https://download.csdn.net/download/weixin_43521269/12837110 人脸识别,是基于人的脸部特征信息 ...

  9. python人脸识别系统界面设计_基于卷积神经网络的人脸识别系统的设计(Python)

    基于卷积神经网络的人脸识别系统的设计(Python)(论文10000字,外文翻译,参考代码,流程图,人脸图像库) 摘要:随着社会的进步与发展,个人信息的保护变得十分重要.传统的密码保护方式已经不再满足 ...

最新文章

  1. 文字超过省略_从楚篆到楚玺的文字结构
  2. More Effective C++读书笔记(二)
  3. 以太坊经典逆势上涨,行情能否继续?
  4. 快速提升网站收录量的技巧有哪些?
  5. (转)explorer.exe应用程序错误:0x000000该内存不能为read的解决方法
  6. HTML和JavaScript代码分离、平稳退化(1)
  7. 计算机如何学会自动地进行图像美学增强?
  8. EIGRP中非等价负载均衡的实现
  9. 【数字逻辑设计】推气泡
  10. Linux基础之常用命令篇
  11. 软件加密狗破解思路和方法
  12. Anaconda如何更新pip
  13. atv320说明书_施耐德变频器参数设置ATV320学习资料
  14. 佳能A系列数码相机(A720 IS)拍摄技巧大全
  15. php文件如何转换成mp4,qlv文件如何转换成mp4 怎样把下载好的qlv格式视频转换成MP4格式...
  16. 梦幻服务器最新开服时间,梦幻服务器开区及合服时间汇总查询(2)
  17. shell 发送短信
  18. 带tan计算机在线,tan等于(万能计算器在线使用)
  19. 这几个习惯,让我成为了高阶网络工程师。
  20. 精美黑色UI版创客新零售系统源码

热门文章

  1. 数据库查询速度优化技巧及解决方案
  2. 传输线的物理基础(三):传输线的瞬时阻抗
  3. 如何搭建Git服务器
  4. C语言实现猜拳小游戏
  5. JAVA基础 第二篇:阶乘
  6. 走近橄榄球运动·体育项目
  7. Java使用quartz实现任务调度定时任务
  8. 正版sql服务器,SQL Server2014企业版 授权许可 正版sql数据库
  9. 最新网站生成APP源代码+Flutter项目/带控制端
  10. FineReport构建银行金融租赁考核系统