一种基于方向梯度直方图与改进胶囊网络的目标检测方法

技术领域

本发明涉及目标检测技术领域，具体涉及一种基于方向梯度直方图与改进胶囊网络的目标检测方法。

背景技术

传统的卷积神经网络通过卷积操作提取目标特征，反向传播实现网络学习，以达到目标检测的目的，在目标检测任务中获得了令人瞩目的成果。然而，传统卷积神经网络存在着图像中元素相对位置、方向等信息的关注不足和池化层信息丢失的问题，障碍物遮挡、恶劣天气等因素都会对检测识别目标造成严重影响。胶囊神经网络提出了胶囊这一概念替代部分卷积神经网络中的神经元，将传统神经网络的标量扩展为向量，有效克服了传统神经网络的不足。因此，将胶囊神经网络应用于目标检测有着更高的整体识别率，胶囊神经网络有着更高的准确率和鲁棒性来满足不同影响因素下的目标检测任务。

然而，传统胶囊网络在给定的位置上只有一个给定类型的胶囊，因此如果一个胶囊网络彼此之间太接近，就不能检测到同一类型的两个对象。陈立潮在《梯度直方图卷积特征的胶囊网络在交通监控下的车型分类》一文中提出采用方向梯度直方图对原始图像进行预处理，在一定程度上解决了无法检测到同一类型的两个对象的问题，但是由于没有改进胶囊网络本身结构，存在原始图像信息提取不够丰富、胶囊结构存在冗余、算法复杂度较高等问题。

发明内容

本发明的目的在于解决现有技术原始图像信息提取不够丰富、胶囊结构存在冗余、算法复杂度较高的技术问题，而提供一种基于方向梯度直方图与改进胶囊网络的目标检测方法。

一种基于方向梯度直方图与改进胶囊网络的目标检测方法，包括以下步骤：

1)获得目标原始图像，使用标注工具标注目标位置，然后随机选取不同的图像作为训练集；

2)将原始图像的方向梯度直方图(HOG)与卷积特征图并联融合以结合图像的边缘轮廓特征与卷积核的视野特征，再以此作为改进胶囊网络的输入；

3)改进胶囊网络利用并行卷积网络提取综合特征，通过去冗胶囊网络形成特征向量，并利用反卷积图像重构网络实现图像重构；

4)利用3*3*256的卷积层和两个并行的1*1卷积核提取检测框中心点特征图和检测框尺度特征图，形成对应目标框并输出检测结果。

优选地，所述步骤2)方向梯度直方图(HOG)与卷积特征图并联融合算法具体步骤为：

2-1)归一化；首先对目标图像划分为4个单元格，对其中的每个单元格划分为9块，并进行Gamma归一化处理，同时对Gamma校正值进行参数寻优；归一化公式如下：

该式中τ表示块的特征向量，ε取值为较小的常数，

2-2)从归一化后的图像中选择检测窗口。选择与图像纵横比相等并且不超过图像大小二分之一的检测窗口。

2-3)从窗口中选择块。依据检测窗口选取长宽相等的矩形块。

2-4)在块内划分cell单元。在矩形块内使用8*8像素大小的正方形cell作为块内特征提取最小单位对块进行划分。

2-5)在cell内进行方向投影。在cell内划分9个方向，每20°为一角度范围提取方向信息。方向信息通过在x水平方向和y垂直方向上将灰度图像I与梯度模板U做卷积操作得到，数学公式如下：

G

G

该式中H(x,y)表示对应坐标下的灰度值，G

2-6)在cell内进行归一化。在cell内统计各个方向角度范围的实际方向角度数量，得到方向直方图，选取角度方向最为集中的方向角度作为cell的方向。

2-7)在块内构建HOG特征。在块内统计各个cell的方向角度范围的实际方向角度数量，得到方向直方图，选取角度方向最为集中的方向角度作为块的方向。

2-8)如果未到达最后一个块，则返回步骤2-3)。

2-9)如果未到达最后一个窗口，则返回步骤2-2)，否则得到方向梯度直方图。

2-10)单行卷积。输入原始图像，使用一层单行卷积层对原始图像进行特征提取，得到卷积特征图。

2-11)特征拼接，并联融合。将卷积特征图与方向梯度直方图进行维度连接，将两种特征图在第三个维度连接，得到28*28*1的图像。

传统的卷积神经网络中采用的池化层结构并没有考虑空间上的相对关系，从而造成这一层的部分价值信息损失。为了解决这一问题，在胶囊网络中使用数字胶囊层来执行池化层的功能，提出了向量神经元，向量神经元将方向、位置等信息通过向量形式存储起来，并在网络中将这些信息不断传递，使得胶囊网络能够对于图像中元素的位置、方向信息的变化具有敏感性。

本发明提出的基于方向梯度直方图的改进胶囊网络并行通过卷积层与HOG特征提取，其图像预处理部分采用HOC-C特征，在保留了原始胶囊网络的优点的同时，方向梯度直方图的引入加强了对于目标检测图像边缘特征信息的提取，通过梯度方向与幅值增加了过于接近的两个胶囊网络之间的辨识度，在一定程度上解决了无法检测到同一类型的两个对象的问题。

优选地，所述步骤3)改进胶囊网络算法具体步骤为：

3-1)利用并行卷积网络提取综合特征。

3-2)利用去冗胶囊网络生成特征向量。

3-3)利用反卷积图像重构网络还原原始图像，评估网络损失。

优选地，所述步骤3-1)并行卷积网络算法具体步骤为：

3-1-1)首先使用并行卷积神经网络作为特征提取网络。将并联融合后的图像作为输入，图像大小为28*28*1。并行卷积神经网络在卷积层采用4个卷积核，卷积核大小分别为3、5、7、9，卷积核个数选定为32个，步长为2。

3-1-2)边界填充。调节padding大小对原矩阵进行边界填充。

3-1-3)特征提取。特征提取层的非线性函数采用PReLU函数，其数学公式如下：

PReLU(x)＝max(0,x)+α*min(0,x)

该式中，α为学习率。

3-1-4)特征张量连接。第三个维度实现特征张量的连接，得到14*14*128的特征张量。

优选地，所述步骤3-2)去冗胶囊网络算法具体步骤为：

3-2-1)输入去冗余。采用并行卷积网络的输出作为去冗主胶囊网络的输入，使用1*1的卷积核去除冗余胶囊使得经过特征提取后16*16的特征图转化为14*14的特征图像，精简胶囊数量至196个。

3-2-2)输入胶囊标量u

3-2-3)输入胶囊向量与变换矩阵相乘得到的向量

该式中W

3-2-4)将向量

3-2-5)使用非线性函数对s

该式中s

3-2-6)使用softmax方程更新耦合系数c

该式中

3-2-7)若对v

优选地，所述步骤3-3)反卷积图像重构网络算法具体步骤为：

3-3-1)图像输入。输入图像的大小为14*14，特征输入为5*5，卷积核大小分别为3、5、7、9，卷积核个数选定为32个，步长为2，通过调节填充使得输出图像为28*28。输入与反卷积之后的输出图像大小公式如下：

该式中

3-3-2)封装信息拆分。对于胶囊中的6272个神经元，通过全连接层转化为14*14*32的张量，将该张量与并行卷积网络中的连接在第三个维度相组合使得14*14*32的张量变为14*14*160的张量。

3-3-3)去冗余反卷积。对14*14*160的张量使用1*1的卷积核进行去冗，最终得到14*14*32的张量，与并行卷积层中相对应的卷积核的进行反卷积操作，生成最终的重构图像。

优选地，所述步骤4)提取检测框中心点特征图和检测框尺度特征图，形成对应目标框并输出检测结果具体步骤为：

4-1)图像输入。以重构图像作为大小为3*3、输出通道为256的卷积层的输入。

4-2)提取中心特征图。以1*1卷积核提取检测框中心点特征图，记对象中心点的坐标标记为正值，而非对象中心点坐标标记为负值。

4-3)提取尺度特征图。以并行的1*1卷积核提取检测框尺度特征图。尺度特征图中的尺度即为检测框的长和宽。

4-4)输出图像。融合中心特征图与尺度特征图完成对象目标检测。

本发明的有益效果在于：

本发明针对现有技术原始图像信息提取不够丰富的问题，对胶囊网络结构本身进行改进，在主胶囊层前增加了并行卷积网络层，并行卷积网络经过卷积特征提取后，将原始图像的卷积特征图与方向梯度直方图两种图像特征信息，直接以维度向量连接的方式而不是网络的方式融合起来，这样使得网络的输入信息更为丰富且更为贴近原始图像信息。

本发明针对现有胶囊结构存在冗余，算法复杂度较高的问题，对主胶囊进行了去冗余的操作。去冗主胶囊网络精简了网络结构，相较于原主胶囊网络的参数有所减少，并使得网络训练所需时间得到优化，提高了算法效率。

附图说明

图1为本发明中目标检测模型的整体结构示意图；

图2为本发明中的方向梯度直方图(HOG)与卷积特征图并联模型流程图；

图3为本发明中的改进胶囊网络架构图；

图4为本发明中的并行卷积网络示意图；

图5为本发明中的去冗胶囊网络示意图；

图6为本发明中的反卷积图像重构网络示意图；

图7为原始图像；

图8为HOG直方处理图像示意图；

图9为行人图像采用本方法检测结果示意图。

具体实施方式

下面对本发明技术方案进行详细说明，但是本发明的保护范围不局限于所述实施例。

如图1所示，一种基于方向梯度直方图与改进胶囊网络的目标检测方法，主要包括以下步骤：

1)获得目标原始图像，如图7所示。

2)如图2，将原始图像的方向梯度直方图与卷积特征图并联融合以结合图像的边缘轮廓特征与卷积核的视野特征，再以此作为改进胶囊网络的输入；

2-1)归一化。首先对目标图像划分为4个单元格，对其中的每个单元格划分为9块，并进行Gamma归一化处理，同时对Gamma校正值进行参数寻优；归一化公式如下：

该式中τ表示块的特征向量，ε取值为较小的常数，

2-2)，从归一化后的图像中选择检测窗口。选择与图像纵横比相等并且不超过图像大小二分之一的检测窗口。

2-3)，从窗口中选择块。依据检测窗口选取长宽相等的矩形块。

2-4)，在块内划分cell单元。在矩形块内使用8*8像素大小的正方形cell作为块内特征提取最小单位对块进行划分。

2-5)，在cell内进行方向投影。在cell内划分9个方向，每20°为一角度范围提取方向信息。方向信息通过在x水平方向和y垂直方向上将灰度图像I与梯度模板U做卷积操作得到，数学公式如下：

G

G

该式中H(x,y)表示对应坐标下的灰度值，G

2-6)在cell内进行归一化。在cell内统计各个方向角度范围的实际方向角度数量，得到方向直方图，如图8所示。选取角度方向最为集中的方向角度作为cell的方向。

2-7)在块内构建HOG特征。在块内统计各个cell的方向角度范围的实际方向角度数量，得到方向直方图，选取角度方向最为集中的方向角度作为块的方向。

2-8)如果未到达最后一个块，则返回步骤3)。

2-9)如果未到达最后一个窗口，则返回步骤2)，否则得到方向梯度直方图。

2-10)单行卷积。输入原始图像，使用一层单行卷积层对原始图像进行特征提取，得到卷积特征图。

2-11)特征拼接，并联融合。将卷积特征图与方向梯度直方图进行维度连接，将两种特征图在第三个维度连接，得到28*28*1的图像。

3)如图3、4，改进胶囊网络利用并行卷积网络提取综合特征，通过去冗胶囊网络形成特征向量，并利用反卷积图像重构网络实现图像重构；

3-1)利用并行卷积网络提取综合特征；

3-1-1)首先使用并行卷积神经网络作为特征提取网络。将并联融合后的图像作为输入，图像大小为28*28*1。并行卷积神经网络在卷积层采用4个卷积核，卷积核大小分别为3、5、7、9，卷积核个数选定为32个，步长为2。

3-1-2)边界填充。调节padding大小对原矩阵进行边界填充。

3-1-3)特征提取。特征提取层的非线性函数采用PReLU函数，其数学公式如下：

PReLU(x)＝max(0,x)+α*min(0,x)

该式中，α为学习率。

3-1-4)特征张量连接。第三个维度实现特征张量的连接，得到14*14*128的特征张量。

3-2)如图5，利用去冗胶囊网络生成特征向量；

3-2-1)输入去冗余。采用并行卷积网络的输出作为去冗主胶囊网络的输入，使用1*1的卷积核去除冗余胶囊使得经过特征提取后16*16的特征图转化为14*14的特征图像，精简胶囊数量至196个。

3-2-2)输入胶囊标量u

3-2-3)输入胶囊向量与变换矩阵相乘得到的向量

该式中W

3-2-4)将向量

3-2-5)使用非线性函数对s

该式中s

3-2-6)使用softmax方程更新耦合系数c

该式中

3-2-7)若对v

3-3)如图6，利用反卷积图像重构网络还原原始图像。

3-3-1)图像输入。输入图像的大小为14*14，特征输入为5*5，卷积核大小分别为3、5、7、9，卷积核个数选定为32个，步长为2，通过调节填充使得输出图像为28*28。输入与反卷积之后的输出图像大小公式如下：

该式中

3-3-2)封装信息拆分。对于胶囊中的6272个神经元，通过全连接层转化为14*14*32的张量，将该张量与并行卷积网络中的连接在第三个维度相组合使得14*14*32的张量变为14*14*160的张量。

3-3-3)去冗余反卷积。对14*14*160的张量使用1*1的卷积核进行去冗，最终得到14*14*32的张量，与并行卷积层中相对应的卷积核的进行反卷积操作，生成最终的重构图像。

4)利用3*3*256的卷积层和两个并行的1*1卷积核提取检测框中心点特征图和检测框尺度特征图，形成对应目标框并输出检测结果。

4-1)图像输入。以重构图像作为大小为3*3、输出通道为256的卷积层的输入。

4-2)中心特征图。以1*1卷积核提取检测框中心点特征图，记对象中心点的坐标标记记为正值，而非对象中心点坐标标记记为负值。

4-3)尺度特征图。以并行的1*1卷积核提取检测框尺度特征图。尺度特征图中的尺度即为检测框的长和宽。

4-4)输出图像。融合中心特征图与尺度特征图完成对象目标检测，如图9所示。