基于嵌入式系统的多任务人脸属性估计算法
DOI:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391.41; TP183

基金项目:

国家自然科学基金(11664005);贵州省科技计划项目(黔科合基础[2020]1Y021);贵州省领域文献的科学知识图谱构建研究(黔教合YJSCXJH [2020]120)。


Multi-tasking face attribute estimation algorithm based on embedded system
Author:
Affiliation:

Fund Project:

National Natural Science Foundation of China, Grant (Nos. 11664005) ;Science and technology planning project of Guizhou province, Grant (No. 2020-1Y021)

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对传统人脸属性估计算法算力大、推理速度慢、精度低,难以完成算法在移动或嵌入式设备上集成应用等问题,提出一种基于嵌入式系统的多任务人脸属性估计算法。首先,采用MobileFaceNet网络中的瓶颈结构融合跨阶段融合网络 (cross stage partial network,CSPNet)和空间金字塔网络 (spatial pyramid pooling network,SPPNet) 设计CSPSPP_bk结构作为人脸属性估计算法共享网络特征提取模块;然后,在局部属性中增加通道注意力机制,在较困难的全局属性中使用更深、性能更优的网络模型作为Teacher模型指导所设计的轻量级多任务属性网络进行知识蒸馏,采用逐层剪枝的方法对网络模型进行优化,优化后的模型量仅1.8 MB;最后,通过动态类别抑制损失函数进行损失度量,均衡样本数据分布。在公共数据集CelebA和Adience数据集上进行测试比较,性别和眼镜的平均准确率分别为98.89%、99.72%,标准差为3.01%时,年龄估计精度为60.21%,在RK3288开发板上的前传推理速度为138 fps。结果表明:所提方法可广泛应用于嵌入式设备和移动边缘设备。

    Abstract:

    Traditional face attribute estimation algorithms have large computing power, slow reasoning speed and low accuracy, which make it difficult to integrate the algorithm into mobile or embedded devices. A multi-task face attribute estimation algorithm based on embedded system was proposed. Firstly, the bottleneck structure of MobileFaceNet network was integrated with cross stage partial network (CSPNet) and spatial pyramid pooling network (SPPNet) to design the CSPSPP_BK structure as the face attribute estimation algorithm to share the network feature extraction module. Then, in the local properties increase channel attention mechanism, using the network in a difficult global properties deeper and better performance of network model as a guide for the proposed model, the design of lightweight and multi-tasking attribute network knowledge distillation, the method of layered pruning was adopted to optimize the network model, the optimized model of quantity is only 1.8 MB. Finally, the dynamic category inhibition loss function was used to measure the loss and equalize the distribution of sample data. Compared with CelebA and Adience data sets, the average accuracy of gender and glasses is 98.89% and 99.72%, respectively. When the standard deviation is 3.01%, the accuracy of age classification is 60.21%, and the pretransmission reasoning speed on RK3288 development board is 138 fps. The results show that the proposed method can be widely applied to embedded devices and mobile edge devices.

    参考文献
    相似文献
    引证文献
引用本文

孙收余,吴凤娇,罗子江,等. 基于嵌入式系统的多任务人脸属性估计算法[J]. 科学技术与工程, 2022, 22(8): 3228-3235.
Sun Shouyu, Wu Fengjiao, Luo Zijiang, et al. Multi-tasking face attribute estimation algorithm based on embedded system[J]. Science Technology and Engineering,2022,22(8):3228-3235.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2021-06-22
  • 最后修改日期:2022-02-25
  • 录用日期:2021-11-16
  • 在线发布日期: 2022-03-21
  • 出版日期:
×
律回春渐,新元肇启|《科学技术与工程》编辑部恭祝新岁!
亟待确认版面费归属稿件,敬请作者关注