| Abstrakt: |
The intelligent classroom is a popular application scenario in the field of artificial intelligence. This paper proposes a face detection algorithm based on improved YOLOv5, named YOLOv5-SASA, to address the issues of missed or false detection caused by small or occluded faces in images captured by cameras located far away or at an angle. The algorithm consists of three parts. Firstly, the CSPDarknet53 network is utilized in the backbone layer, and the BasicRFB module is used in the final spatial pooling layer to enhance the network's feature extraction ability. Secondly, the NWD loss function is employed to improve the model's robustness in detecting small targets. Thirdly, the independent self- attention mechanism module SASA is introduced in the head layer to address the issue of face occlusion and reduce the model's parameter count. Finally, the improved YOLOv5 network is optimized by reducing the number of neurons in the middle layer channels and adjusting the learning rate to avoid overfitting. Experimental results demonstrate that the proposed method outperforms the original network in the easy, medium, and hard levels of the WiderFace validation set, achieving accuracies of 97.5%, 96.3%, and 86.5%, respectively, which effectively improves the accuracy of face detection in classroom scenarios. [ABSTRACT FROM AUTHOR] |
