基于ViT的面部表情识别模型,在FER2013、MMI和AffectNet数据集上微调,支持七种情绪分类
下载量 8,730
发布时间 : 4/29/2024
模型简介
该模型是基于视觉Transformer(ViT)架构的面部情绪识别模型,专门针对七种基本情绪进行分类,包括愤怒、厌恶、恐惧、高兴、悲伤、惊讶和中性。
模型特点
多数据集融合训练
结合FER2013、MMI和AffectNet三个面部表情数据集进行训练,提高模型泛化能力
高效视觉Transformer架构
采用ViT基础架构,通过16x16图像块处理,在224x224分辨率下实现高效特征提取
优化的训练策略
使用余弦退火学习率调度和预热策略,配合Adam优化器实现稳定训练
模型能力
面部情绪识别
七种基本情绪分类
静态图像情感分析
使用案例
人机交互
情感感知系统
用于智能设备的情感感知界面,根据用户表情调整交互方式
准确率84.34%
心理健康
情绪状态监测
辅助心理医生或护理人员监测患者的情绪变化
精选推荐AI模型
Llama 3 Typhoon V1.5x 8b Instruct
专为泰语设计的80亿参数指令模型,性能媲美GPT-3.5-turbo,优化了应用场景、检索增强生成、受限生成和推理任务
大型语言模型
Transformers

支持多种语言
L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-Tiny是一个基于SODA数据集训练的超小型对话模型,专为边缘设备推理设计,体积仅为Cosmo-3B模型的2%左右。
对话系统
Transformers

英语
C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
基于RoBERTa架构的中文抽取式问答模型,适用于从给定文本中提取答案的任务。
问答系统
中文
R
uer
2,694
98
AIbase是一个专注于MCP服务的平台,为AI开发者提供高质量的模型上下文协议服务,助力AI应用开发。
简体中文