买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：南京信息工程大学

摘要：本发明公开了一种基于增强特征空间对齐融合网络的人群计数方法，包括：获取人群图像数据集并进行数据预处理，将人群图像数据集划分成训练集、验证集和测试集，利用标注图产生人群密度图；构建增强特征空间对齐融合网络；利用训练集、验证集对增强特征空间对齐融合网络进行训练，获得最优网络模型；将测试集输入到最优网络模型，得到人群图像计数结果。本发明能够充分利用人群计数任务中特征图的空间信息进行特征增强和特征对齐，密度图质量更高，计数准确率更高。

主权项：1.一种基于增强特征空间对齐融合网络的人群计数方法，其特征在于，所述人群计数方法包括以下步骤：S1：获取人群图像数据集并进行数据预处理，将人群图像数据集划分成训练集、验证集和测试集，利用标注图产生人群密度图；S2：构建增强特征空间对齐融合网络，所述增强特征空间对齐融合网络包括特征提取网络、三个空间可分离特征增强模块、两个空间对齐特征融合模块和密度图生成模块；所述增强特征空间对齐融合网络使用VGG-16网络的前13个卷积层和前4个池化层作为特征提取网络，对人群图像中的特征进行提取；所述三个空间可分离特征增强模块分别对第3个最大池化层、第4个最大池化层和第13个卷积层输出的特征图C1、特征图C2和特征图C3进行特征增强，获得特征图S1、特征图S2和特征图S3；其中一个空间对齐特征融合模块将作为浅层特征的特征图S2和作为深层特征的特征图S3进行特征融合，输出特征图A1；另一个空间对齐特征融合模块将作为浅层特征的特征图S1和作为深层特征的特征图A1进行特征融合，输出特征图A2；所述密度图生成模块对特征图C2和特征图A2进行处理，输出与输入图像相同尺寸的人群密度图，再对人群密度图进行积分运算得到人群数量；S3：利用步骤S1得到的训练集、验证集对增强特征空间对齐融合网络进行训练，获得最优网络模型；S4：将步骤S1得到的测试集输入到步骤S3得到的最优网络模型，得到人群图像计数结果；所述空间可分离特征增强模块将输入的特征图送入三个空间可分离卷积分支中进行卷积处理，三个空间可分离卷积分支的卷积核大小分别设置为5、7、9；接着将三个空间可分离卷积分支的输出直接相加，相加结果经过3×3卷积层卷积后分别送入三个通道分支，在第一个通道分支依次进行最大池化、1×1卷积、Relu激活和1×1卷积操作，在第二个通道分支依次进行平均池化、1×1卷积、Relu激活和1×1卷积操作；然后将第一个通道分支和第二个通道分支输出的特征图相加后，使用Sigmoid激活得到通道注意力权重；再将第三个分支的特征图与通道注意力权重相乘，得到空间可分离特征增强模块的输出特征图；所述空间对齐特征融合模块对输入的深层特征进行上采样使其尺寸与浅层特征相同，接着将上采样的结果与浅层特征进行拼接，拼接后的特征通过一个1×1卷积和一个输出通道数为4的3×3卷积，接着将输出结果分别与浅层特征和经过上采样的深层特征相加，其输出结果再分别输入一个3×3卷积，然后将输出特征使用配准函数R通过插值的方式对不同深度的特征图进行空间对齐，将对齐后的不同深度的特征图进行相加得到初步融合特征；同时将浅层特征分别进行平均池化和最大池化，将两个池化后的结果拼接后进行7×7的卷积操作，接着使用Sigmoid激活函数进行激活生成空间注意力权重，将初步融合特征与空间注意力权重相乘，得到空间对齐特征融合模块输出的融合特征。

全文数据：

权利要求：

百度查询： 南京信息工程大学 基于增强特征空间对齐融合网络的人群计数方法