Small face detection based on improved YOLOv5s
At present, in the complex real-world application scenairos, the task of small face detection encounters numerous challenges, which include small face scale, abrupt lighting changes and low accuracy. In order to address the concerns of overlooking small face detection within existing models, this st...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | zho |
| Published: |
POSTS&TELECOM PRESS Co., LTD
2024-12-01
|
| Series: | 智能科学与技术学报 |
| Subjects: | |
| Online Access: | http://www.cjist.com.cn/zh/article/doi/10.11959/j.issn.2096-6652.202438/ |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1832586375346520064 |
|---|---|
| author | ZHOU Lifang HU Zhen LIU Bo |
| author_facet | ZHOU Lifang HU Zhen LIU Bo |
| author_sort | ZHOU Lifang |
| collection | DOAJ |
| description | At present, in the complex real-world application scenairos, the task of small face detection encounters numerous challenges, which include small face scale, abrupt lighting changes and low accuracy. In order to address the concerns of overlooking small face detection within existing models, this study introduced a novel small face detection model termed SK-YOLOv5s which was based on convolutional kernel attention mechanism. Firstly, we proposed a small face enhancement module to fuse and upsample multi-layer features, which enhances the resolution of small face feature maps and strengthens their distinctiveness. Subsequently, we incorporated the SKNet attention mechanism into the model, which can adaptively adjust receptive field sizes across multiple scales and enhance the detection efficacy of small face. Finally, EIoU was utilized as the loss function, which directly reduced the width and height discrepancies between predicted and actual bounding boxes, and FReLU was utilized as the activation function, which could enhance the nonlinear expressiveness of feature maps to improve the precision and stability of small face detection. The performance of the enhanced model on the WIDER FACE dataset demonstrated mean average precision improvement of 7.9% over YOLOv5s. The experimental results demonstrate the viability of the enhanced model for small face detection in real-world scenarios. |
| format | Article |
| id | doaj-art-de434c5afac542d8b786b69e42ee5003 |
| institution | Kabale University |
| issn | 2096-6652 |
| language | zho |
| publishDate | 2024-12-01 |
| publisher | POSTS&TELECOM PRESS Co., LTD |
| record_format | Article |
| series | 智能科学与技术学报 |
| spelling | doaj-art-de434c5afac542d8b786b69e42ee50032025-01-25T19:00:51ZzhoPOSTS&TELECOM PRESS Co., LTD智能科学与技术学报2096-66522024-12-01645646581046393Small face detection based on improved YOLOv5sZHOU LifangHU ZhenLIU BoAt present, in the complex real-world application scenairos, the task of small face detection encounters numerous challenges, which include small face scale, abrupt lighting changes and low accuracy. In order to address the concerns of overlooking small face detection within existing models, this study introduced a novel small face detection model termed SK-YOLOv5s which was based on convolutional kernel attention mechanism. Firstly, we proposed a small face enhancement module to fuse and upsample multi-layer features, which enhances the resolution of small face feature maps and strengthens their distinctiveness. Subsequently, we incorporated the SKNet attention mechanism into the model, which can adaptively adjust receptive field sizes across multiple scales and enhance the detection efficacy of small face. Finally, EIoU was utilized as the loss function, which directly reduced the width and height discrepancies between predicted and actual bounding boxes, and FReLU was utilized as the activation function, which could enhance the nonlinear expressiveness of feature maps to improve the precision and stability of small face detection. The performance of the enhanced model on the WIDER FACE dataset demonstrated mean average precision improvement of 7.9% over YOLOv5s. The experimental results demonstrate the viability of the enhanced model for small face detection in real-world scenarios.http://www.cjist.com.cn/zh/article/doi/10.11959/j.issn.2096-6652.202438/Small Face Detectionattention mechanismYOLOv5s |
| spellingShingle | ZHOU Lifang HU Zhen LIU Bo Small face detection based on improved YOLOv5s 智能科学与技术学报 Small Face Detection attention mechanism YOLOv5s |
| title | Small face detection based on improved YOLOv5s |
| title_full | Small face detection based on improved YOLOv5s |
| title_fullStr | Small face detection based on improved YOLOv5s |
| title_full_unstemmed | Small face detection based on improved YOLOv5s |
| title_short | Small face detection based on improved YOLOv5s |
| title_sort | small face detection based on improved yolov5s |
| topic | Small Face Detection attention mechanism YOLOv5s |
| url | http://www.cjist.com.cn/zh/article/doi/10.11959/j.issn.2096-6652.202438/ |
| work_keys_str_mv | AT zhoulifang smallfacedetectionbasedonimprovedyolov5s AT huzhen smallfacedetectionbasedonimprovedyolov5s AT liubo smallfacedetectionbasedonimprovedyolov5s |