1.一种基于改进YOLOv5的实时小人脸检测方法,其特征在于,包括以下步骤:步骤1:将人脸数据集划分为训练集和测试集,对训练集进行包括以下预处理:数据集的清洗:检查数据集中的图片和标签文件是否完整,否则删除多余的图片和标签文件,同时将长宽比太小的锚框删除,并且在删除后,检查下是否存在空的标注文件,如果有,也删除;
根据Kmeans聚类算法得到自适应锚框;
步骤2:基于YOLOv5的小人脸检测模型,引入SK选择性卷积核注意力,通过动态计算每个卷积核得到通道的权重,动态的将各个卷积核的结果进行融合;
步骤3:设计小人脸特征增强模块,小人脸特征增强模块是:在第17层后,继续对特征图进行上采样处理,使得特征图继续扩大,同时第20层获取到的特征图与第2层的特征图进行融合(concat),以获得更大的特征图进行小目标的检测,将特征图进行上采样处理,使得特征图继续扩大,同时将浅层特征图与深层特征图进行融合;
步骤4:采用YOLOv5模型,并将主干特征提取网络C3替换为ShuffleNetV2网络;
步骤5:将训练好的小人脸检测深度学习模型应用于真实场景的小人脸检测。
2.根据权利要求1所述的一种基于改进YOLOv5的实时小人脸检测方法,其特征在于,所述步骤1:将人脸数据集划分为训练集和测试集,对训练集进行预处理,根据聚类算法得到自适应锚框,具体包含以下步骤:A1、对训练集中的图片数据进行数据增强,通过平移、旋转、改变图片的宽高比、调整饱和度在内的操作,增加样本数据;
A2、通过聚类算法,对训练集中的真实锚框进行聚类分析,得到精确的锚框作为小人脸检测网络的预定值。
3.根据权利要求2所述的一种基于改进YOLOv5的实时小人脸检测方法,其特征在于,所述A2、通过聚类算法,对训练集中的真实锚框进行聚类分析,得到精确的锚框作为小人脸检测网络的预定值,具体包括:yolov5生成anchors中不仅仅使用了k‑means聚类,还使用了Genetic Algorithm遗传算法,在k‑means聚类的结果上进行mutation变异;
1.读取训练集中每张图片的宽高以及所有锚框的宽高;
2.将每张图片中宽高的最大值等比例缩放到指定大小img_size;
3.将锚框从相对坐标改成绝对坐标即乘以缩放后的宽高;
4.筛选锚框,保留宽高都大于等于两个像素的锚框;
5.使用k‑means聚类得到n个锚框;
6.使用遗传算法随机对锚框的宽高进行变异,如果变异后效果变得更好,就将变异后的结果赋值给锚框,如果变异后效果变差就跳过,默认变异1000次;
7.将最终变异得到的锚框返回。
4.根据权利要求1所述的一种基于改进YOLOv5的实时小人脸检测方法,其特征在于,所述步骤2:基于YOLOv5的小人脸检测模型,引入SK注意力,通过动态计算每个卷积核得到通道的权重,动态的将各个卷积核的结果进行融合,具体包括以下步骤:H×W×c
B1、对于任何给定的特征图X∈R ,H、W、c分别表示图像的宽、高、通道数,R表示特征图,经过两次变换 和 和 均由深度卷积组成,归一化和激活函数依次运行;传统的卷积5*5被空洞卷积替换扩张为2,卷积核大小为3*
3;不同分支不同的卷积核处理输入X特征,得到多分支的处理后的特征图;
B2、控制来自多个分支的信息流携带不同尺度的信息进入下一层的神经元;首先通过融合来自多个分支的结果: 然后U经过Fgp全局平均池化Fgp、i、j、uc(i,j)分别表示全局空间信息、通道的高从1到H、通道的宽从1到W、通道统计信息,H*W表示空间维度。
5.根据权利要求4所述的一种基于改进YOLOv5的实时小人脸检测方法,其特征在于,所述引入SK注意力,通过动态计算每个卷积核得到通道的权重其中 Fscale(uc,sc)指的是标量sc和特征图的逐通道乘法。
6.根据权利要求1所述的一种基于改进YOLOv5的实时小人脸检测方法,其特征在于,所述步骤3:设计了小人脸特征增强模块,使得特征图继续扩大,同时将浅层特征图与深层特征图进行融合,具体包括以下步骤:设计了小人脸特征增强模块,在第17层后,继续对特征图进行上采样处理,使得特征图继续扩大,同时第20层获取到的特征图与第2层的特征图进行融合(concat),以获得更大的特征图进行小目标的检测;
C1、将特征图进行上采样处理,使得特征图扩大;采用最近邻插值的方法进行上采样处理;
C2、将浅层特征图与深层特征图进行融合,浅层网络提取的特征和输入比较近,包含更多的像素点的信息,一些细粒度的信息是图像的一些颜色、纹理、边缘、棱角信息,但是由于经过的卷积更少,其语义性更低,噪声更多;深层网络提取的特征,是图像整体性的一些信息,一些粗粒度的信息,包含是更抽象的信息,即语义信息;在第17层后,继续对特征图进行上采样处理,使得特征图继续扩大,同时第20层获取到的特征图与第2层的特征图进行融合(concat),以获得更大的特征图进行小目标的检测。
7.根据权利要求1所述的一种基于改进YOLOv5的实时小人脸检测方法,其特征在于,所述步骤4:采用YOLOv5模型,并将主干特征提取网络C3替换为ShuffleNetV2网络,具体步骤为:YOLOv5模型除了ShuffleNetV2网络,还包括CBL、Resunit、SPP、Focus。
8.根据权利要求1所述的一种基于改进YOLOv5的实时小人脸检测方法,其特征在于,所述步骤5将小人脸检测深度学习模型应用于真实场景的小人脸检测,具体包括:
1.利用图形化工具‘Auto Py To Exe’,将训练好的best.pt文件导入到weights文件夹,生成exe文件;
2.双击运行exe文件,支持图片检测、视频检测、摄像头检测。