面部表情识别(Pytorch):人脸检测模型+面部表情识别分类模型

目录

  • [0 相关资料](#0 相关资料)
  • [1 基于人脸检测+面部表情分类识别方法](#1 基于人脸检测+面部表情分类识别方法)
  • [2 项目安装](#2 项目安装)
    • [2.1 平台与镜像](#2.1 平台与镜像)
    • [2.2 项目下载](#2.2 项目下载)
    • [2.3 模型下载](#2.3 模型下载)
    • [2.4 上传待测试图片](#2.4 上传待测试图片)
    • [2.5 项目安装](#2.5 项目安装)
  • [3 demo测试](#3 demo测试)

0 相关资料

面部表情识别2:Pytorch实现表情识别(含表情识别数据集和训练代码):https://blog.csdn.net/guyuealian/article/details/129505205

b站视频:https://www.bilibili.com/video/BV1xm4y1p7H3

1 基于人脸检测+面部表情分类识别方法

项目源码:https://github.com/Whiffe/PyTorch-Facial-Expression-Recognition

面部表情识别由两部分组成:人脸检测与表情识别分类

人脸检测:https://github.com/Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB

表情识别分类: resnet18,resnet34,resnet50, mobilenet_v2以及googlenet等常见的深度学习模型

2 项目安装

2.1 平台与镜像

我是实用的AutoDL平台

镜像选择:

PyTorch 1.7.0

Python 3.8(ubuntu18.04)

Cuda 11.0

2.2 项目下载

项目下载:

复制代码
git clone https://github.com/Whiffe/PyTorch-Facial-Expression-Recognition.git

如果网速问题无法下载,我已经同步到码云上(推荐)

复制代码
git clone https://gitee.com/YFwinston/PyTorch-Facial-Expression-Recognition.git

2.3 模型下载

模型权重下载(latest-model-099-94.7200.pth ):https://download.csdn.net/download/WhiffeYF/88196455

将下载的模型放在:

PyTorch-Facial-Expression-Recognition/data/pretrained/mobilenet_v2_1.0_CrossEntropyLoss_20230313090258/model

模型权重下载(rfb-face-mask.pth ): https://download.csdn.net/download/WhiffeYF/88196487

将下载的模型放在:

PyTorch-Facial-Expression-Recognition/libs/light_detector/data/pretrained/pth

2.4 上传待测试图片

在该目录下,上传待检测图片:

PyTorch-Facial-Expression-Recognition/data/test_image

2.5 项目安装

在 PyTorch-Facial-Expression-Recognition 目录下执行:

复制代码
pip install -r requirements.txt 

3 demo测试

复制代码
python demo.py --image_dir data/test_image --model_file data/pretrained/mobilenet_v2_1.0_CrossEntropyLoss_20230313090258/model/latest_model_099_94.7200.pth --out_dir output/

检测结果如下:






相关推荐
腾视科技-AI7 小时前
腾视科技大模型一体机解决方案:低成本私有化落地,重塑行业智能应用新格局
大数据·人工智能·科技·ai·ai大模型·腾视科技·ai算力盒
Thomas.Sir7 小时前
第26课:TensorFlow|循环神经网络RNN原理【时序数据处理、序列依赖关系讲解】
人工智能·rnn·tensorflow
Wang's Blog7 小时前
Vibe Coding一人即团队系列54:云服务器 Node.js 与 MySQL 9 环境搭建及配置指南
服务器·人工智能·mysql·node.js
测试开发Kevin7 小时前
DeepEval + Eval‑Harness 完整讲解(结合 Playwright UI 自动化例子)
人工智能·ai·langchain
张欣-男7 小时前
5分钟理解线性代数v2
人工智能·线性代数·机器学习
pnoker7 小时前
从工业软件到 AI 智能体:工业 AIoT 技术路线的系统梳理
java·人工智能·物联网·microsoft·开源·工业互联网
2501_931819707 小时前
车载语音青岛方言数据集搭建阶段 信实翻译垂直场景标准赋能青岛方言标注公司作业
人工智能·语音识别
jay神7 小时前
深度学习如何确定Batch Size的大小?
人工智能·深度学习·计算机视觉·毕业设计·课程设计·batch
Geeys7 小时前
拼多多商家客户端和网页版有哪些功能上的差异?
大数据·人工智能
Rina GO7 小时前
PS怎么把文字贴在圆柱体上?2种实用方法教程附不规则物体贴图
人工智能·贴图·ps插件·不规则物体贴图·文字贴图·ai贴图