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 分钟前
探索 Codex:AI 编程助手的未来潜力
人工智能·openai·代码生成·codex·ai编程助手
没事儿写两篇6 分钟前
Python 包管理工具-uv
python·uv·开源包管理工具
2501_941418558 分钟前
基于YOLO11-C3k2-ESC的避雷器外部缺陷检测实现
python
流㶡9 分钟前
Python爬虫:POST与Selenium
爬虫·python·selenium
菜青虫嘟嘟9 分钟前
Expert Iteration:一种无需人工标注即可显著提升大语言模型推理能力的简单方法核心
人工智能·语言模型·自然语言处理
玄同76515 分钟前
LangChain v1.0+ Retrieval模块完全指南:从文档加载到RAG实战
人工智能·langchain·知识图谱·embedding·知识库·向量数据库·rag
爱学习的阿磊16 分钟前
持续集成/持续部署(CI/CD) for Python
jvm·数据库·python
deepdata_cn21 分钟前
为什么AI需要因果?
人工智能·因果学习
说私域32 分钟前
社群招募文案的核心构建要点与工具赋能路径——基于AI智能名片链动2+1模式商城小程序的实践研究
人工智能·小程序·私域运营