04 面部表情识别:Pytorch实现表情识别-表情数据集训练代码

总目录:人脸检测与表情分类
https://blog.csdn.net/whiffeyf/category_12793480.html

目录

  • [0 相关资料](#0 相关资料)
  • [1 面部表情识数据集](#1 面部表情识数据集)
  • [2 模型下载](#2 模型下载)
  • [3 训练](#3 训练)

0 相关资料

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

开始前需要安装:
01 面部表情识别(Pytorch):人脸检测模型+面部表情识别分类模型

1 面部表情识数据集

类别文件 PyTorch-Facial-Expression-Recognition/data/class_name.txt(最后一行,请多回车一行)

修改配置文件的数据路径:​config.yaml​



2 模型下载

mobilenet-v2-b0353104, resnet18-5c106cde, resnet34-333f7ec4三个模型的下载

mobilenet-v2-b0353104 resnet18-5c106cde resnet34-333f7ec4 预训练模型

下载后放在:PyTorch-Facial-Expression-Recognition/classifier/models/pretrained/

3 训练

可能需要安装

bash 复制代码
source /etc/network_turbo
bash 复制代码
pip install sklearn
pip install tensorboardX

训练

复制代码
python train.py -c configs/config.yaml 


训练结果在:

PyTorch-Facial-Expression-Recognition/work_space/mobilenet_v2_1.0_CrossEntropyLoss_20240924_175800_1880/model/best_model_098_94.8200.pth

相关推荐
陈天伟教授22 分钟前
基于学习的人工智能(7)机器学习基本框架
人工智能·学习
千里念行客2401 小时前
昂瑞微正式启动科创板IPO发行
人工智能·科技·信息与通信·射频工程
习习.y1 小时前
python笔记梳理以及一些题目整理
开发语言·笔记·python
撸码猿1 小时前
《Python AI入门》第10章 拥抱AIGC——OpenAI API调用与Prompt工程实战
人工智能·python·aigc
qq_386218991 小时前
Gemini生成的自动搜索和下载论文的python脚本
开发语言·python
vx_vxbs661 小时前
【SSM电影网站】(免费领源码+演示录像)|可做计算机毕设Java、Python、PHP、小程序APP、C#、爬虫大数据、单片机、文案
java·spring boot·python·mysql·小程序·php·idea
双翌视觉2 小时前
双翌全自动影像测量仪:以微米精度打造智能化制造
人工智能·机器学习·制造
编程小白_正在努力中2 小时前
神经网络深度解析:从神经元到深度学习的进化之路
人工智能·深度学习·神经网络·机器学习
无风听海3 小时前
神经网络之经验风险最小化
人工智能·深度学习·神经网络
音视频牛哥3 小时前
轻量级RTSP服务的工程化设计与应用:从移动端到边缘设备的实时媒体架构
人工智能·计算机视觉·音视频·音视频开发·rtsp播放器·安卓rtsp服务器·安卓实现ipc功能