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

相关推荐
哆啦叮当1 分钟前
VADv2 基于概率规划的端到端自动驾驶模型
人工智能·机器学习·自动驾驶
五月底_6 分钟前
GRPO参数详解
人工智能·深度学习·nlp·rl·grpo
沃达德软件6 分钟前
大数据治安防控中心
大数据·人工智能·信息可视化·数据挖掘·数据分析
雾江流9 分钟前
肉包 1.4.0 | 豆包AI手机平替,开源免费,AI自动化
运维·人工智能·自动化·软件工程
光锥智能10 分钟前
昆仑芯冲刺IPO,百度押中了一枚国产AI芯片
人工智能·百度
沫儿笙12 分钟前
发那科弧焊机器人保护气节气设备
人工智能·机器人
hopsky20 分钟前
经典Transformer的PyTorch实现
pytorch·深度学习·transformer
有Li22 分钟前
AGFS-Tractometry:一种新型图谱引导的精细尺度束测量方法,用于增强扩散MRI束描记术的沿束组统计比较|文献速递-医疗影像分割与目标检测最新技术
人工智能
中科米堆27 分钟前
中科米堆CASAIM自动化三维检测-0.02mm计量级精度产品尺寸快速检测
人工智能·3d·3d全尺寸检测
张拭心34 分钟前
为什么说 AI 视频模型不能用来做教育?Sora-2 Veo-3 来了也不行
前端·人工智能