Imagen de Portada

Study on an Improved YOLOv7-Based Algorithm for Human Head Detection

Guardado en:
Publicado en:Electronics vol. 14, no. 9 (2025), p. 1889
Autor principal: Wu, Dong
Otros Autores: Yan, Weidong, Wang, Jingli
Publicado:
MDPI AG
Materias:
Feature extraction
Accuracy
Public spaces
Cameras
Labels
Collaboration
Deep learning
Frames (data processing)
Computer vision
Pedestrians
Artificial neural networks
Frames per second
Shopping centers
Inference
Algorithms
Video data
Modules
Surveillance
Object recognition
Localization
Crowds
Efficiency
Acceso en línea:Citation/Abstract
Full Text + Graphics
Full Text - PDF
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
Resumen:In response to the decreased accuracy in person detection caused by densely populated areas and mutual occlusions in public spaces, a human head-detection approach is employed to assist in detecting individuals. To address key issues in dense scenes—such as poor feature extraction, rough label assignment, and inefficient pooling—we improved the YOLOv7 network in three aspects: adding attention mechanisms, enhancing the receptive field, and applying multi-scale feature fusion. First, a large amount of surveillance video data from crowded public spaces was collected to compile a head-detection dataset. Then, based on YOLOv7, the network was optimized as follows: (1) a CBAM attention module was added to the neck section; (2) a Gaussian receptive field-based label-assignment strategy was implemented at the junction between the original feature-fusion module and the detection head; (3) the SPPFCSPC module was used to replace the multi-space pyramid pooling. By seamlessly uniting CBAM, RFLAGauss, and SPPFCSPC, we establish a novel collaborative optimization framework. Finally, experimental comparisons revealed that the improved model’s accuracy increased from 92.4% to 94.4%; recall improved from 90.5% to 93.9%; and inference speed increased from 87.2 frames per second to 94.2 frames per second. Compared with single-stage object-detection models such as YOLOv7 and YOLOv8, the model demonstrated superior accuracy and inference speed. Its inference speed also significantly outperforms that of Faster R-CNN, Mask R-CNN, DINOv2, and RT-DETRv2, markedly enhancing both small-object (head) detection performance and efficiency.
ISSN:2079-9292
DOI:10.3390/electronics14091889
Fuente:Advanced Technologies & Aerospace Database