Research on Small Face Detection Algorithm Based on Lightweight Intra-Group Attention Mechanism and Multi-Scale Feature Fusion
摘要
With the advancement of mobile device hardware capabilities and the rapid progress of artificial intelligence technologies, the focus of research has shifted towards face detection technology on mobile devices, particularly in efficiently and accurately detecting small faces characterized by low resolution and susceptibility to occlusion. This paper addresses this challenge by introducing an algorithm named CA-YOLOv8n, which integrates a lightweight intra-group attention mechanism with an adaptive feature fusion strategy to enhance the detection performance of small faces. Initially, a lightweight attention mechanism module called CRIAM is proposed to amplify features of small faces while mitigating background interference. Subsequently, an Adaptive Feature Fusion module (AFF) is developed to dynamically adjust its fusion strategy based on facial features of varying sizes, thereby increasing detection precision. Ultimately, experiments conducted on the WiderFace dataset indicate that compared to the benchmark algorithm YOLOv8n, the proposed algorithm achieves a 7.1% improvement in mAP50 and a 3.9% improvement in mAP50-95.