首页|基于改进YOLOv5s的小人脸检测

基于改进YOLOv5s的小人脸检测

扫码查看
目前在复杂的真实应用场景中,小人脸检测任务面临诸多挑战,如人脸尺度小、光照变化剧烈和精度较低等。针对现有模型容易忽视小人脸检测的问题,提出了一种基于卷积核注意力机制的小人脸检测模型SK-YOLOv5s。首先,设计一种小人脸增强模块,将浅层特征与深层特征进行融合与上采样,提高了小人脸特征图的分辨率,从而增强其特征;其次,在模型中引入SKNet注意力机制,能够多尺度自适应调节感受野大小,提高小人脸的检测率;最后,采用EIoU作为目标损失函数、FReLU作为激活函数,直接减小预测与真实边界框之间的宽度和高度差异,增强特征图的非线性表达能力,提高小人脸检测的精度和稳定性,相较于YOLOv5s,所改进模型在WIDER FACE数据集中的mAP提升了0。079。实验结果证明了所改进模型在真实场景下进行小人脸检测的可行性。
Small face detection based on improved YOLOv5s
At present,in the complex real-world application scenairos,the task of small face detection encounters numer-ous 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 pre-dicted 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 en-hanced 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.

small face detectionattention mechanismYOLOv5s

周丽芳、胡振、刘波

展开 >

重庆邮电大学软件工程学院,重庆 400065

小人脸检测 注意力机制 YOLOv5s

2024

智能科学与技术学报

智能科学与技术学报

CSTPCD
ISSN:
年,卷(期):2024.6(4)