基于多尺度轻量化网络的头部姿态估计方法、系统、设备及介质

摘要

本申请提供一种基于多尺度轻量化网络的头部姿态估计方法、系统、设备及介质，包括：获取包含头部姿态的数据集，预处理数据集；利用多尺度卷积网络提取预处理的数据集得到相应的特征图；基于所述特征图训练轻量化网络得到MobileNet回归器模型；获取待测图像的头部图像，将其输入到MobileNet回归器模型进行头部姿态预测，得到所述待测图像的头部姿态信息。本发明采用多尺度卷积核提取数据集内特征图，使用不同尺度的卷积核对输入头部姿态图像提取特征，丰富了图像特征，保留图像信息，提高了头部姿态估计的准确率；同时，基于轻量化网络训练MobileNet回归器模型，在不损失网络性能的前提下，大幅度降低了计算量。