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

相关推荐
豌豆花下猫5 分钟前
Python 潮流周刊#71:PyPI 应该摆脱掉它的赞助依赖(摘要)
后端·python·ai
Tech Synapse13 分钟前
Vscode 远程切换Python虚拟环境
ide·vscode·python
大模型实战1 小时前
深入探索《AI大模型开发之路》:我的读书心得
人工智能
颜问儿1 小时前
【python】横截面数据分析及可视化报告示例
python·信息可视化·数据分析
sp_fyf_20241 小时前
计算机前沿技术-人工智能算法-大语言模型-最新研究进展-2024-09-25
人工智能·深度学习·算法·语言模型·自然语言处理
红米煮粥1 小时前
卷积神经网络-最优模型
人工智能·深度学习·cnn
赫连达1 小时前
大模型输入参数学习
人工智能·学习·ai·大模型
Strengthen_沈1 小时前
简单分享下Python的if
开发语言·python
"Return"1 小时前
美团2024年秋招第一场笔试算法题题解【技术】
java·c++·python·算法
DA树聚1 小时前
人工智能-机器学习-深度学习-分类与算法梳理
人工智能·深度学习·算法·机器学习·语言模型·分类·transformer