4 个月前

学习基于多维边缘特征的面部动作单元关系图以实现面部动作单元识别

学习基于多维边缘特征的面部动作单元关系图以实现面部动作单元识别

摘要

面部动作单元(AUs)的激活状态相互影响。尽管一对AUs之间的关系可能复杂且独特,现有的方法未能为每个面部表情中的每对AUs具体且明确地表示这些线索。本文提出了一种AU关系建模方法,通过深度学习生成一个独特的图,以明确描述目标面部表情中每对AUs之间的关系。我们的方法首先将每个AU的激活状态及其与其他AUs的关联编码为节点特征。然后,它学习一组多维边特征,以描述每对AUs之间多个任务特定的关系线索。在节点和边特征学习过程中,我们的方法还考虑了特定面部表情对AUs关系的影响,通过将整个面部表示作为输入。在BP4D和DISFA数据集上的实验结果表明,节点和边特征学习模块显著提升了基于CNN和变压器的骨干网络的性能,我们的最佳系统达到了最先进的AU识别结果。我们的方法不仅在建模AU识别的关系线索方面具有强大的能力,还可以轻松集成到各种骨干网络中。我们已开源了PyTorch代码。

代码仓库

tomas-gajarsky/facetorch
pytorch
GitHub 中提及
cvi-szu/me-graphau
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
facial-action-unit-detection-on-bp4dResNet 50
Average F1: 59.1
facial-action-unit-detection-on-bp4dMulti-dimensional Edge Feature-based AU Relation Graph (Swin-B)
Average AUC: 83.1
Average F1: 65.5
facial-action-unit-detection-on-bp4dMulti-dimensional Edge Feature-based AU Relation Graph (ResNet 50)
Average AUC: 82.6
Average F1: 64.7
facial-action-unit-detection-on-bp4dSwin-B
Average F1: 62.6
facial-action-unit-detection-on-disfaMulti-dimensional Edge Feature-based AU Relation Graph (ResNet 50)
Average AUC: 92.9
Average F1: 63.1
facial-action-unit-detection-on-disfaMulti-dimensional Edge Feature-based AU Relation Graph (Swin-B)
Average AUC: 92.1
Average F1: 62.4

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
学习基于多维边缘特征的面部动作单元关系图以实现面部动作单元识别 | 论文 | HyperAI超神经