YOLO-FaceV2: A scale and occlusion aware face detector

In recent years, face detection algorithms based on deep learning have made great progress. Nevertheless, the effective utilization of face detectors for small and occlusion faces remains challenging, primarily stemming from the limitations in pixel information and the presence of missing features....

Celý popis

Uložené v:
Podrobná bibliografia
Vydané v:Pattern recognition Ročník 155; s. 110714
Hlavní autori: Yu, Ziping, Huang, Hongbo, Chen, Weijun, Su, Yongxin, Liu, Yahui, Wang, Xiuying
Médium: Journal Article
Jazyk:English
Vydavateľské údaje: Elsevier Ltd 01.11.2024
Predmet:
Face detection
Imbalance problem
Occlusion
Scale-aware
YOLO
YOLO
Occlusion
Imbalance problem
Face detection
Scale-aware
ISSN:0031-3203, 1873-5142
On-line prístup:Získať plný text
Tagy: Pridať tag
Žiadne tagy, Buďte prvý, kto otaguje tento záznam!
Popis
Shrnutí:In recent years, face detection algorithms based on deep learning have made great progress. Nevertheless, the effective utilization of face detectors for small and occlusion faces remains challenging, primarily stemming from the limitations in pixel information and the presence of missing features. In this paper, we propose a novel real-time face detector, YOLO-FaceV2, built upon the YOLOv5 architecture. Our approach introduces a Receptive Field Enhancement (RFE) module designed to extract multi-scale pixel information and augment the receptive field for accurately detecting small faces. To address issues related to face occlusion, we introduce an attention mechanism termed the Separated and Enhancement Attention Module (SEAM), which effectively focuses on the regions affected by occlusion. Furthermore, we propose a Slide Weight Function (SWF) to mitigate the imbalance between easy and hard samples. The experiments demonstrate that our YOLO-FaceV2 achieves performance exceeding the state-of-the-art on the WiderFace validation dataset. Source code and pre-trained model are available at https://github.com/Krasjet-Yu/YOLO-FaceV2. •Proposed an YOLO-FaceV2 detector to address face detection.•Good performance under face occlusion and varying scales.•Designed a novel weighting function alleviated the problem of imbalanced samples.•Detection results on the WiderFace validation dataset are 98.6%, 97.9% and 91.9%.•Achieved state-of-the-art performance on the easy and medium subset of WiderFace dataset.
ISSN:0031-3203
1873-5142
DOI:10.1016/j.patcog.2024.110714