公开(公告)号：CN117953570A

公开(公告)日：2024-04-30

申请号：CN202410216429.8

申请日：2024-02-27

申请人： 西安电子科技大学

发明人： 王笛 ,  李欣浩 ,  刘锦辉 ,  王泉 ,  万波 ,  罗雪梅 ,  王义峰

IPC分类号： G06V40/16 ,  G06V10/42 ,  G06V10/44 ,  G06V10/774 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/084

摘要： 本发明公开了一种基于协同学习的头部姿态估计方法，主要解决现有技术估计精度较低，前向推理时间较长的问题。其实现方案为：获取人脸图像数据集，划分训练数据和测试数据；构建包括人脸关键点热力图模块、头部姿态初步预测模块和姿态角模块的头部姿态估计网络，在这些模块中共享人脸特征图，在人脸关键点和头部姿态角之间进行协同学习；将训练数据输入到对头部姿态估计网络采用反向传播法对其训练；将测试数据输入训练好的头部姿态估计网络得到头部姿态估计结果。本发明通过共享人脸特征图，缩短了在多人脸场景下头部姿态估计时间，可有效地提高估计效率，并通过协同学习提高了头部姿态角的估计精度，可用于疲劳驾驶检测、智慧课堂行为分析、人机交互。