[发明专利]图像分类卷积神经网络结构的构建方法

摘要

一种图像分类卷积神经网络结构的构建方法，由构建卷积神经网络结构、卷积神经网络训练和测试步骤组成。构建卷积神经网络结构的步骤为：获得训练样本图像和测试样本图像并对其进行预处理、设定卷积神经网络初始结构、引入激活函数、确定池化方法、确定滤波器、确定卷积层个数，得到图像分类卷积神经网络结构；卷积神经网络的训练和测试步骤为：用K‑Means聚类方法获得含有20个特征图的卷积层的滤波器，经前向传播和反向传播更新网络权值矩阵至最大训练次数，得到训练好的卷积神经网络，对测试样本图像进行测试，得到验证后的图像分类卷积神经网络结构，具有构建方法简单、有效等优点，适用于门牌号码、手写数字、邮政编码等图像分类。

主权项

1.一种图像分类卷积神经网络结构的构建方法，由构建卷积神经网络结构、卷积神经网络训练和测试步骤组成，其特征在于所述的构建卷积神经网络结构步骤如下：(1)获得训练样本图像和测试样本图像并对其进行预处理(a)从图像数据集中选取50400～73257张训练样本图像和6000～26032张测试样本图像；(b)对50400～73257张训练样本图像进行预处理包括图像灰度化、图像减均值步骤：图像灰度化：用(1)式对被训练的样本图像进行灰度化处理，将彩色图像转为灰度图像：I＝W1×R+W2×G+W3×B                      (1)式中I为灰度化后的图像，R为彩色图像的红色分量，G为彩色图像的绿色分量，B为彩色图像的蓝色分量，W1为彩色图像的红色分量的权重，W2为彩色图像的绿色分量的权重，W3为彩色图像的蓝色分量的权重；图像减均值：用(2)式对已被灰度化的训练样本图像进行减均值处理：式中Z为减均值后的图像，vi为n个图像中的第i个图像的像素矩阵，n为50400～73257整数个图像；(c)对6000～26032张测试样本图像进行预处理测试样本图像的预处理与对训练样本进行预处理步骤(b)相同；(2)设定卷积神经网络初始结构设定卷积神经网络初始结构：输入层(1)是50400～73257张大小为28×28或32×32的训练样本图像，输入层(1)后连接含有20个特征图的卷积层(2)，含有20个特征图的卷积层(2)后连接含有20个特征图的池化层(3)，含有20个特征图的池化层(3)后连接含有64个特征图的卷积层(2)，含有64个特征图的卷积层(2)后连接含有64个特征图的池化层(3)，含有64个特征图的池化层(3)后连接含有500个特征图的卷积层(2)，含有500个特征图的卷积层(2)后连接全连接层(4)，全连接层(4)后连接输出层(5)；(3)引入激活函数在每个卷积层(2)后引入ReLu激活函数为：yi＝max{xi,0}                              (3)式中xi为与前一层输出相连的第i个权值矩阵与当前层输入的内积，即卷积结果，yi为第i个特征图的输出，20≤i≤500；(4)确定池化方法卷积神经网络中的池化层(3)选用最大池化方法或均值池化方法或随机池化方法进行池化；(5)确定滤波器在3×3～9×9方阵之间、以图像宽度28或32为步长，在20～500内进行网格搜索，得到滤波器，为卷积层(2)的权值矩阵，滤波器的大小在3×3～9×9方阵之间，个数在20～500之间；(6)确定卷积层(2)个数在含有20个特征图的池化层(3)后增加一层卷积层(2)，其滤波器大小和个数与含有20个特征图的卷积层(2)的滤波器大小和个数相同，通过式(4)得到该卷积层(2)特征图的大小，在含有64个特征图的池化层(3)后增加或不增加一层卷积层(2)，若增加一层卷积层，其滤波器大小和个数与含有64个特征图的卷积层(2)的滤波器大小和个数相同，通过式(4)得到该卷积层(2)特征图的大小，卷积层(2)个数为3～5层，选择使得卷积神经网络对所选测试样本图像6000～26032张分类效果最好时的卷积层(2)个数，得到图像分类卷积神经网络结构：o＝I*－m+1                                  (4)式中I*为本层输入矩阵的长，m为滤波器的像素矩阵长，o为卷积层(2)特征图的像素矩阵长；所述的卷积神经网络的训练和测试步骤为：(1)用K‑Means聚类方法获得含有20个特征图的卷积层(2)的权值矩阵(a)从训练集中随机选取N个m×m矩阵大小的图像块，N为所选取图像块有限的正整数，m为含有20个特征图的卷积层(2)的滤波器的像素矩阵长，m为3～9；(b)用式(5)对步骤(a)中所选取的N个图像块中的每一个图像块进行归一化处理：式中x*为m×m矩阵大小的图像块，μ为N个图像块的平均值，σ为N个图像块的标准差，x表示经过归一化后的结果；(c)以式(6)为目标函数，对上步得到的图像块进行10次K‑Means聚类方法，取其中目标函数值最小的一次所得到的20个聚类中心，作为含有20个特征图的卷积层(2)的滤波器初始化值：式中k为20是聚类个数，s表示图像块的类别数，x_j为第i类别中的图像块，为第i类的样本均值，i为1～20的整数；(2)将预处理后的50400～73257张训练样本图像输入图像分类卷积神经网络结构，经过前向传播和反向传播两个步骤反复循环更新网络的权值矩阵，直至达到30～70次数，训练过程结束，得到训练好的卷积神经网络；(3)将预处理后的6000～26032张测试样本图像输入到训练好的卷积神经网络进行测试，得到验证后的图像分类卷积神经网络结构。