一种基于深度Gabor网络的图像目标识别方法与流程
本发明涉及图像处理技术，特别是涉及一种基于深度gabor网络的图像目标识别方法。
背景技术：
针对图像的目标识别可以用于各方面，包括网络上图像目标的理解、智慧城市、监控场景中的目标的发掘、人机交互中的场景理解等。深度学习技术给图像目标的识别提供了强有力的手段，并在多个社会领域上取得了成功。包括基于图像内容的搜索、以图搜图、监控中的人物再识别等。目前面向图像目标识别的深度学习主要采用基于卷积神经网络的结构，并在结构、网络的层数、网络的参数量上逐渐提高，对系统提高了要求。为了将网络的参数量降低，研究一种小参数量的网络具有重要意义。
现有图像目标识别的网络采用卷积核作为基本的结构，由于该结构靠大量的参数的学习来拟合训练数据，其确实理论的解释也是被大家诟病的地方。本发明创新性的提出采用gabor滤波器来提取图像的特征，由于gabor滤波器模拟人脑对于纹理的响应，有理论分析与支撑。并提出紧接gabor层的激活层、批标准化层和池化层，完成参数的学习和训练。
技术实现要素：
有鉴于此，本发明的主要目的在于提供一种基于gabor算子的目标识别方法，并且能相比现有方法具有更小的参数量与适应范围。
为了达到上述目的，本发明提出的技术方案为：一种基于深度gabor网络的图像目标识别方法，包括如下步骤：
步骤1、读取需要进行目标识别的图像或视频数据，如果数据为视频数据，则视频数据分解成单帧图像序列；将图像数据，或者视频分解出的单帧图像序列，分为训练数据和测试数据，分别构成训练数据集与测试数据集。
步骤2、在目标识别的算法中包含训练和测试两个过程，在训练过程中，将对于步骤1得到的训练数据进行标记，根据图像中目标种类对图像进行标记，将图像数据打上类别标签，得到标记后的训练数据。
步骤3、针对步骤2得到的标记后的训练数据进行数据增广，将增广后的数据加到原始数据中，得到的数据增广后的训练数据。
步骤4、将步骤3得到的数据增广后的训练数据作为深度gabor网络的输入，利用监督学习方法，通过最小化损失函数，训练深度gabor网络，得到深度gabor网络。利用gabor网络对训练数据进行分类，对比数据标签，得到分类正确的训练数据与分类错误的训练数据，分类错误的训练数据被认为是困难数据。
步骤5、对于步骤4得到的深度gabor网络，针对分类错误的训练数据进行困难数据挖掘，采用生成对抗网络生成类似的困难数据，将困难数据通过深度gabor网络进行训练，提升深度gabor网络的性能，至此完成训练阶段。
步骤6、在测试阶段中，利用训练好的深度gabor网络作为测试过程的网络，对于输入的图像利用gabor网络提取图像的特征，输出对该图像数据的识别结果，从而对图像目标识别。
本发明与现有技术相比的优点在于：
(1)本发明具有模型可解释的特点。由于传统的深度学习都是采用卷积核来对图像进行处理，并没有很强的理论的解释性。本发明由于采用gabor滤波器作为基础来构建神经网络，利用gabor模拟人的视觉感知中对于纹理与边界敏感的特点，具有可解释的理论基础。
(2)本发明具有参数量小的特点。由于采用gabor滤波器，每个的滤波器的参数量相比卷积核的参数小。在相同的层数与识别准确率的要求下，本发明所提的网络参数量小。
(3)本发明具有是用范围广的特点。本发明所提的网络具有参数量小的特点，而且具有可解释性。所以具体的任务其能有针对性的调整，且能部署在服务器已经算力小的端上，具有广泛的应用范围。
附图说明
图1为本发明深度gabor网络结构示意图。
具体实施方式
下面结合附图以及具体实施方式进一步说明本发明。
如图1所示，本发明一种用于图像目标识别的基于gabor网络，具体实现步骤如下：
步骤1、读取需要进行目标识别的图像或视频数据，如果数据为视频数据，则视频数据分解成单帧图像序列；将图像数据，或者视频分解出的单帧图像序列，分为训练数据和测试数据，分别构成训练数据集与测试数据集。
步骤2、在目标识别的算法中包含训练和测试两个过程，在训练过程中，将对于步骤1得到的训练数据进行标记，根据图像中目标种类对图像进行标记，将图像数据打上类别标签，得到标记后的训练数据。
步骤3、针对步骤2得到的标记后的训练数据进行数据增广，将增广后的数据加到原始数据中，得到的数据增广后的训练数据。
步骤i)针对步骤2得到的标记后的训练数据，利用图像旋转、局部裁剪、上下对称翻转、左右对称翻转、颜色抖动方式进行图像数据增广。
步骤ii)对数据增广后得到的更多的数据，加到原始数据中，增大训练数据集的规模，提高神经网络的训练的有效性。
步骤iii)图像目标识别的训练过程，为一个多分类问题，利用标签对数据进行分类，对于数据扩展后的标签，其值与进行变换前标签相同。
步骤4、将步骤3得到的数据增广后的训练数据作为深度gabor网络的输入，利用监督学习方法，通过最小化损失函数，训练深度gabor网络，得到深度gabor网络。利用gabor网络对训练数据进行分类，对比数据标签，得到分类正确的训练数据与分类错误的训练数据，分类错误的训练数据被认为是困难数据。
步骤i)训练过程为有监督学习过程，图像目标识别的数据为多类别的数据，神经网络采用交叉熵作为损失函数：
loss＝-y′logy
式中y为预测结果，y’为实际结果。并通过随机梯度下降法完成对于深度gabor网络参数的调整。
步骤ii)对于图像目标识别的深度gabor网络的组成均为gabor滤波器，其表示如下：
x′＝xcosθ+ysinθ
y′＝-xsinθ+ycosθ
其中x,y为二维卷积核坐标，λ为波长，θ为滤波器方向，σ为高斯核标准差。
步骤iii)深度gabor网络核心为4个gabor层，每个gabor层由多个不同参数的gabor滤波器组成，gabor滤波器中θ与λ为可训练参数，随机初始化取值如下：
θ∈(-π,π)
λ∈(2,0.2×min(height,width))
height与width为输入图像长和宽。
步骤iv)深度gabor网络具体结构如下：
(1)input：输入层为原始数据；
(2)gabor1：本层针对不同的输入通道采用相同gabor滤波器，滤波器数量取64；
(3)p1：本层为池化层，采用步长为2，大小为2*2的最大池化，可提高网络的训练速度并避免过拟合现象；
(4)s1：本层为激活层，采用“relu”函数作为激活函数，可有效避免过拟合现象；
(5)n1：本层为批标准化层，其用于将本层的输入归一化为均值为0，标准差为1的数据，这对避免过拟合现象有很好的效果，其计算方法如下，
yi＝γx′i+β
其中m为每批样本个数，则显然μb为样本平均值，
(6)gabor2：本层针对不同的输入通道采用相同gabor滤波器，滤波器数量取256；
(7)p2：本层为池化层，具体同p1；
(8)s2：本层为激活层，具体同s1；
(9)n2：本层为批标准化层，具体同n1；
(10)gabor3：本层针对不同的输入通道采用相同gabor滤波器，滤波器数量取256；
(11)p3：本层为池化层，具体同p1；
(12)s3：本层为激活层，具体同s1；
(13)n3：本层为批标准化层，具体同n1；
(14)d1：本层将数据降维成为一维数据，失去全部结构信息但保留了所有数据信息；
(15)output：输出层与d1层采用全连接，输出层的向量长度应与输入视频信息的动类别数目相等。
步骤v)根据步骤i设定的损失函数loss，取学习率为0.001对网络进行训练，参数优化方法采用梯度下降法。
步骤5、对于步骤4得到的深度gabor网络，针对分类错误的训练数据进行困难数据挖掘，采用生成对抗网络生成类似的困难数据，将困难数据通过深度gabor网络进行训练，提升深度gabor网络的性能，至此完成训练阶段。
步骤i)在目标分类深度gabor神经网络训练过程中，需要对易被分类错的样本进行困难样本挖掘，提高边神经网络分类的准确率。利用生成对抗网络对容易被分错的困难样本进行相似样本的生成。生成对抗网络为无监督学习，生成对抗网络包括生成网络与对抗网络。生成网络根据输入的数据生成与之相似的容易被深度gabor网络分错的数据，根据公式
步骤ii)设定训练步数x，若达到训练要求，则停止针对容易分错样本的对抗生成网络的训练，利用生成的数据作为训练数据集的补充样本；
步骤iii)将容易分错的样本与生成对抗网络生成的样本融合，作为扩充的训练样本数据集，提高图像目标分类的精度，修正深度gabor网络。至此完成训练阶段。
步骤6、在测试阶段中，利用训练好的深度gabor网络作为测试过程的网络，对于输入的图像利用gabor网络提取图像的特征，输出对该图像数据的识别结果，从而对图像目标识别。
步骤i)将输入的图像数据进行预处理。如果输入数据为视频数据，则频分解成单个帧，而后输入训练好的深度gabor网络单帧图像进行测试。如果输入数据为图像，则将单帧图像输入训练好的深度gabor网络单帧图像进行测试。
步骤ii)利用训练好的深度gabor网络作为测试过程的网络，将输入的图像利用gabor网络与输入图像进行gabor滤波提取图像的特征，利用对应数据的点乘提取纹理特征，利用池化进行空间输入融合，得到深度gabor网络针对图像的特征描述。
步骤iii)利用深度gabor网络后端的全连接网络，对深度gabor网络得到到图像特征进行分类，输出对该图像数据的识别结果，从而完成对测试图像的目标识别。
综上所述，本发明所述的一种基于gabor网络的图像目标识别方法，首先对图像中的特征提取，即设计基于gabor滤波器的基本核，提取图像中的纹理等特征；gabor网络的结构设计，在gabor特征核的基础上，设计多层的特征提取网络；为了提高网络的分类的准确率，gabor网络采用困难样本挖掘，利用生成对抗网络来提高网络的性能。将带有对应动作类别信息的数据作为训练数据进行训练，训练好的神经网络可实现对输入数据代表的动作进行判断，由此图像目标识别的智能处理。
提供以上实施例仅仅是为了描述本发明的目的，而并非要限制本发明的范围。本发明的范围由所附权利要求限定。不脱离本发明的精神和原理而做出的各种等同替换和修改，均应涵盖在本发明的范围之内。
- 还没有人留言评论。精彩留言会获得点赞！