基于改进YOLOv5的多任务安全人头检测算法  

Multi-Task Secure Head-Detection Algorithm Based on Improved YOLOv5

作　　者：毛雨晴 赵奎 MAO Yuqing;ZHAO Kui(School of Cyber Science and Engineering,Sichuan University,Chengdu 610065,China)

机构地区：[1]四川大学网络空间安全学院,成都610065

出　　处：《计算机工程》2022年第8期136-143,共8页Computer Engineering

基　　金：国家自然科学基金(61872254)。

摘　　要：针对目前监控摄像头预置检测算法存在人脸隐私泄露风险的问题,提出一种基于YOLOv5结构的多任务安全人头检测算法Privacyface-YOLO及轻量级检测版本Privacyface-YOLO(Light)。Privacyface-YOLO的主干网络采用带残差结构的跨阶段局部网络ResCSP,通过残差结构降低特征学习的复杂度,提升网络对人头特征的提取能力。轻量级版本Privacyface-YOLO(Light)使用深度可分离卷积替换ResCSP模块,以减少计算量并提升网络的运行速度。模型的颈部模块引入特征金字塔与路径聚合网络,实现跨层多尺度信息交流,提升模型在复杂人头与小目标场景下的鲁棒性。对提取出的特征进行三分支输出,其中目标定位分支用于定位人头坐标,人头分割分支生成覆盖人头的掩膜以保证人脸隐私,人数回归分支获得图像中的人头总数,通过多任务三分支结构保护人脸隐私同时满足更加复杂的场景需求。实验结果表明,Privacyface-YOLO算法能够有效完成实时人头检测任务并保护人脸隐私,相较目前主流人头检测算法,在人头数据集HollywoodHeads和Brainwash上,该算法的平均查准率AP50指标分别提升11.8%和5.8%,平均查准率AP70指标分别提升20.2%和35.2%。Considering the risk of privacy violations that could leak facial information in existing detection algorithms used by surveillance cameras,this study proposes a multi-task secure head-detection algorithm based on a YOLOv5 structure,called Privacyface-YOLO,as well as a lightweight version called Privacyface-YOLO(Light).The backbone network of Privacyface-YOLO adopts the cross-stage local network ResCSP with a residual structure,which reduces the complexity of feature learning and improves the ability of the network to extract features.The lightweight version Privacyface-YOLO(Light)replaces the ResCSP module with deep separable convolution to reduce the amount of computation performed and improve running speed.A neck module introduces a feature pyramid and a path aggregation network to realize cross-layer and multi-scale information exchange and to improve robustness to scenes with small targets and complex head shapes.The extracted features are output in three branches that perform different tasks to protect privacy and meet the needs of complex scenes.A target-location branch is used to locate the coordinates of people’s heads in an image,and a head-segmentation branch generates a mask covering each head to ensure the privacy of face information,while a dedicated regression branch obtains the total number of heads in the image.Experimental results are presented to show that the Privacyface-YOLO algorithm can perform the real-time head-detection task effectively while protecting the privacy of face information.The proposed algorithm outperformed existing mainstream head-detection algorithms,obtaining a higher AP50 average precision index on the HollywoodHeads and Brainwash datasets by 11.8%and 5.8%,respectively;its AP70 average precision index was higher by 20.2%and 35.2%,respectively.

关 键 词：人头检测 深度学习 YOLOv5算法 人脸隐私 目标检测 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]