python+pytorch人脸表情识别

概述

基于深度学习的人脸表情识别,数据集采用公开数据集fer2013,可直接运行,效果良好,可根据需求修改训练代码,自己训练模型。

详细

一、概述

本项目以PyTorch为框架,搭建卷积神经网络模型,训练后可直接调用py文件进行人脸检测与表情识别,默认开启摄像头实时检测识别。效果良好,可根据个人需求加以修改。

二、演示效果:

三、实现过程

  1. 搭建网络

    def init(self):
    super(FaceCNN, self).init()

    复制代码
     # 第一次卷积、池化
     self.conv1 = nn.Sequential(
         nn.Conv2d(in_channels=1, out_channels=64, kernel_size=3, stride=1, padding=1),  # 卷积层
         # BatchNorm2d进行数据的归一化处理,这使得数据在进行Relu之前不会因为数据过大而导致网络性能的不稳定
         nn.BatchNorm2d(num_features=64),  # 归一化
         nn.RReLU(inplace=True),  # 激活函数
         nn.MaxPool2d(kernel_size=2, stride=2),  # 最大值池化
     )
    
     # 第二次卷积、池化
     self.conv2 = nn.Sequential(
         nn.Conv2d(in_channels=64, out_channels=128, kernel_size=3, stride=1, padding=1),
         nn.BatchNorm2d(num_features=128),
         nn.RReLU(inplace=True),
         nn.MaxPool2d(kernel_size=2, stride=2),
     )
    
     # 第三次卷积、池化
     self.conv3 = nn.Sequential(
         nn.Conv2d(in_channels=128, out_channels=256, kernel_size=3, stride=1, padding=1),
         nn.BatchNorm2d(num_features=256),
         nn.RReLU(inplace=True),
         nn.MaxPool2d(kernel_size=2, stride=2),
     )
    
     # 参数初始化
     self.conv1.apply(gaussian_weights_init)
     self.conv2.apply(gaussian_weights_init)
     self.conv3.apply(gaussian_weights_init)
    
     # 全连接层
     self.fc = nn.Sequential(
         nn.Dropout(p=0.2),
         nn.Linear(in_features=256 * 6 * 6, out_features=4096),
         nn.RReLU(inplace=True),
         nn.Dropout(p=0.5),
         nn.Linear(in_features=4096, out_features=1024),
         nn.RReLU(inplace=True),
         nn.Linear(in_features=1024, out_features=256),
         nn.RReLU(inplace=True),
         nn.Linear(in_features=256, out_features=7),
     )
  2. 训练模型

    载入数据并分割batch

    train_loader = data.DataLoader(train_dataset, batch_size)

    损失函数

    loss_function = nn.CrossEntropyLoss()

    学习率衰减

    scheduler = optim.lr_scheduler.StepLR(optimizer, step_size=10, gamma=0.8)

    device = "cuda" if torch.cuda.is_available() else 'cpu'

    构建模型

    model = FaceCNN().to(device)

    优化器

    optimizer = optim.SGD(model.parameters(), lr=learning_rate, weight_decay=wt_decay)

    逐轮训练

    for epoch in range(epochs):
    if (epoch + 1) % 10 == 0:
    learning_rate = learning_rate * 0.1
    # 记录损失值
    loss_rate = 0
    # scheduler.step() # 学习率衰减
    model.train() # 模型训练
    for images, labels in train_loader:
    images, labels = images.to(device), labels.to(device)
    # 梯度清零
    optimizer.zero_grad()
    # 前向传播
    output = model.forward(images)
    # 误差计算
    loss_rate = loss_function(output, labels)
    # 误差的反向传播
    loss_rate.backward()
    # 更新参数
    optimizer.step()

  3. 模型预测

    with torch.no_grad():
    pred = model(face)
    probability = torch.nn.functional.softmax(pred, dim=1)
    probability = np.round(probability.cpu().detach().numpy(), 3)
    max_prob = np.max(probability)
    # print(max_prob)
    predicted = classes[torch.argmax(pred[0])]
    cv2.putText(img, predicted + " " + str(max_prob), (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 1, (100, 255, 0), 1, cv2.LINE_AA)
    cv2.imshow('frame', img)

相关推荐
前端付豪1 分钟前
17、自动化才是正义:用 Python 接管你的日常琐事
后端·python
jioulongzi3 分钟前
记录一次莫名奇妙的跨域502(badgateway)错误
开发语言·python
破无差34 分钟前
python实现简单的地图绘制与标记20250705
python
向阳@向远方35 分钟前
第二章 简单程序设计
开发语言·c++·算法
喜欢吃豆1 小时前
目前最火的agent方向-A2A快速实战构建(二): AutoGen模型集成指南:从OpenAI到本地部署的全场景LLM解决方案
后端·python·深度学习·flask·大模型
Mr_Xuhhh1 小时前
信号与槽的总结
java·开发语言·数据库·c++·qt·系统架构
纳兰青华1 小时前
bean注入的过程中,Property of ‘java.util.ArrayList‘ type cannot be injected by ‘List‘
java·开发语言·spring·list
好开心啊没烦恼1 小时前
Python 数据分析:DataFrame,生成,用字典创建 DataFrame ,键值对数量不一样怎么办?
开发语言·python·数据挖掘·数据分析
liulilittle1 小时前
VGW 虚拟网关用户手册 (PPP PRIVATE NETWORK 基础设施)
开发语言·网络·c++·网关·智能路由器·路由器·通信
Devil枫2 小时前
Kotlin高级特性深度解析
android·开发语言·kotlin