政策资讯

图像分类卷积神经网络结构的构建方法

专利类型:
发明
申请号/专利号:
CN201710106457.4
申请人(专利权人):
马苗
行业类别:
G06K9/62
技术成熟度:
公布时间:
2018-10-23
证书状态:
授权
交易价格:
30000元以上
我要咨询

摘要详情

技术摘要

权利要求书

技术附图

交易流程

委托经理人

一种图像分类卷积神经网络结构的构建方法,由构建卷积神经网络结构、卷积神经网络训练和测试步骤组成。构建卷积神经网络结构的步骤为:获得训练样本图像和测试样本图像并对其进行预处理、设定卷积神经网络初始结构、引入激活函数、确定池化方法、确定滤波器、确定卷积层个数,得到图像分类卷积神经网络结构;卷积神经网络的训练和测试步骤为:用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表示图像块的类别数,xj为第i类别中的图像块,为第i类的样本均值,i为1~20的整数;

(2)将预处理后的50400~73257张训练样本图像输入图像分类卷积神经网络结构,经过前向传播和反向传播两个步骤反复循环更新网络的权值矩阵,直至达到30~70次数,训练过程结束,得到训练好的卷积神经网络;

(3)将预处理后的6000~26032张测试样本图像输入到训练好的卷积神经网络进行测试,得到验证后的图像分类卷积神经网络结构。

2.根据权利要求1所述的图像分类卷积神经网络结构的构建方法,其特征在于:所述的构建卷积神经网络结构步骤的确定滤波器步骤(5)中,滤波器的大小分别为9×9、5×5、4×4,以图像宽度32为步长,滤波器的个数分别为20、64、500,得到滤波器,为卷积层(2)的权值矩阵。

3.根据权利要求1所述的图像分类卷积神经网络结构的构建方法,其特征在于所述的卷积神经网络的训练和测试步骤的用K-Means聚类方法获得含有20个特征图的卷积层(2)的权值矩阵步骤(1)中的(a)步骤为:从训练集中随机选取N个m×m矩阵大小的图像块,N为所选取的400000图像块,m为含有20个特征图的卷积层(2)的滤波器的像素矩阵长。


请联系平台。

我要咨询

商标号:
联系人:
联系电话:
商标名称:
报价:
需求描述:
提交
服务
客服
电话:18504815395
邮箱:965848622@qq.com
地址:呼和浩特市赛罕区昭乌达路70号内蒙古科技大厦906
微信
招聘
返回顶部