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

相关推荐
Jaly_W2 分钟前
用于航空发动机故障诊断的深度分层排序网络
人工智能·深度学习·故障诊断·航空发动机
CodeClimb4 分钟前
【华为OD-E卷-木板 100分(python、java、c++、js、c)】
java·javascript·c++·python·华为od
小嗷犬4 分钟前
【论文笔记】Cross-lingual few-shot sign language recognition
论文阅读·人工智能·多模态·少样本·手语翻译
夜幕龙11 分钟前
iDP3复现代码数据预处理全流程(二)——vis_dataset.py
人工智能·python·机器人
吃个糖糖28 分钟前
36 Opencv SURF 关键点检测
人工智能·opencv·计算机视觉
AI慧聚堂41 分钟前
自动化 + 人工智能:投标行业的未来是什么样的?
运维·人工智能·自动化
盛世隐者42 分钟前
【pytorch】循环神经网络
人工智能·pytorch
晚夜微雨问海棠呀44 分钟前
长沙景区数据分析项目实现
开发语言·python·信息可视化
cdut_suye1 小时前
Linux工具使用指南:从apt管理、gcc编译到makefile构建与gdb调试
java·linux·运维·服务器·c++·人工智能·python
开发者每周简报1 小时前
微软的AI转型故事
人工智能·microsoft