公开(公告)号：CN118366205A

公开(公告)日：2024-07-19

申请号：CN202410777240.6

申请日：2024-06-17

Applicant: 长城信息股份有限公司

Inventor： 胡翔 ,  朱秀军 ,  乔伟晨 ,  王波 ,  吴笛 ,  李杰

IPC: G06V40/16 ,  G06N3/0464 ,  G06N3/08 ,  G06V10/80 ,  G06V10/82

Abstract: 本发明涉及一种基于注意力机制的轻量化人脸追踪方法及系统，方法包括：构建预设人脸检测YOLO‑Face模型；使用GSConv对原始特征提取网络进行改进，得到改进特征提取网络；将ASFF网络结构与原始特征融合网络结合，得到改进特征融合网络；将SimAM机制加入原始特征输出网络的检测头中，得到改进特征输出网络；构建得到轻量化人脸追踪模型；通过人脸追踪数据训练集对轻量化人脸追踪模型进行优化训练，得到最优人脸追踪模型；当接收到待追踪图像时，通过最优人脸追踪模型输出人脸追踪结果。通过注意力机制改进轻量化的YOLO‑Face模型，优化了模型结构的同时并通过增加较少的计算量，从而换取人脸追踪效果的提升。