基于elm的图像去噪方法
【技术领域】
[0001]本发明涉及计算机图像处理技术领域，尤其涉及一种基于ELM的快速图像去噪方法。
【背景技术】
[0002]图像去噪是很多图像处理的先决条件，因此，图像修复与去除噪声一直是图像处理领域的热点问题。在图像修复领域，存在的难点问题是去除噪声的同时保留原始的信息结构。为了达到这个目的，数十年来，3类主要的方法被提出以期解决该问题:空域方法，转换域方法以及基于学习的方法。
[0003]空域方法中比较有代表性的是BF(Bilateral filter,双边滤波法)，NLM(non-local means,非局部均值法)。BF利用了临近位置的相关性，以及临近色域的相关性来消除图像噪声；NLM则从图像全局出发，将相似性比较高的patch组合在一起，利用这样一个个的群体的相似性解决问题。BF本身去噪效果有限，而NLM的一个比较大的局限就是需要大量运算，效率不高。
[0004]转换域方法是将图像从空间域转换到其他域如小波域等，通过某些方法得到的阈值以消除图像中的高频噪声，之后再从转换域转换到空间域。该方法对消除噪声效果比较好，但是由于图像细节往往也处在高频部分，因此，该方法在消除噪声的同时也抹去了图像本来具有的一些细节特征。
[0005]通常的方法常常针对一种噪声类型，而基于学习的方法确可以通过学习灵活的应用在多种噪声类型的图像修复任务中。基于学习的方法需要预先训练修复模型(多为网络模型)，从大量样本中获取特征训练得到修复模型，然后将训练后的修复模型应用到待处理图像中。从这个意义上看，该方法算是NLM的泛化，所不同的是基于学习的方法一次训练，可以一直使用训练参数。因此这类方法，训练完成后，时间花费较低，而且效果也不错。但是该方法存在如下缺点:训练过程长，需要花费大量时间训练(数天甚至数个月)。
【发明内容】
[0006]针对现有技术的不足，本发明提供一种基于ELM(超限学习机)的图像去噪方法。
[0007]一种基于ELM的图像去噪方法，包括如下步骤:
[0008](I)根据待处理图像的大小构建基本前馈神经网络；
[0009](2)针对所述的基本前馈神经网络构建训练样本集；
[0010](3)利用所述的训练样本集基于ELM法训练所述的基本前馈神经网络得到训练好的神经网络；
[0011](4)将待处理的图像输入到训练好的神经网络，相应的输出即为去噪后的图像。
[0012]本发明的图像去噪方法是一种基于学习的方法，可以很方便的应用在多种类型噪声的图像修复场景中，在训练时采用定制训练样本集并采用ELM算法进行训练，大大提高了训练速度。
[0013]所述基本前馈神经网络包括3层，分别为输入层、特征抽取层和输出层，输入层的节点数和输出层的节点数均等于待处理图像的像素点的个数。
[0014]输入层接受向量化的待处理图像，中间层(即特征抽取层)进行特征抽取，输出层输出处理结果(即去噪后的图像)，为保证去噪后图像大小不变，输入层与输出层节点数目一致，均等于待处理图像的大小。
[0015]出于简化训练的目的，设计网络结构时尽量简单，3层网络可以使用广义线性方程进行参数求解，而不是使用非常耗时的BP算法进行迭代求解；且具有3层的网络在使用ELM求解时，收敛性上有理论保障。
[0016]所述基本前馈神经网络中，输入层到特征抽取层、特征抽取层到输出层均采用全连接；
[0017]输入层到特征抽取层的各个连接参数在(0，I)之间随机取值，且采用非线性连续可导函数作为激活函数；
[0018]特征抽取层到输出层的所有连接参数均未知。
[0019]在网络连接中有全连接与部分连接的区别，这两种连接方式各有千秋，部分连接通常利用局部相关性降低所需参数，但这是一个比较强的假设，理论上的效果会稍差，因此，理论上说全连接方式会取得更好的方法，但由于参数比部分连接要多，训练更困难耗时更多，因此训练未必能够达到最优的效果。本发明中利用elm中收敛性的理论，随机分配第一层参数后不再调整，仅对第二层的连接参数进行调整，这样的话就不需要调整第一层的连接参数，大大降级了训练时的计算量，且采用全连接方式同时也保证了训练效果。
[0020]作为优选，输入层到特征抽取层采用sigmoid函数作为激活函数。
[0021]所述步骤⑵具体如下:
[0022]获取若干个大小与待处理图像相同原始图像，并对各个原始图像添加噪声以形成训练样本，所有训练样本的集合即为网络训练集。
[0023]本发明中的原始图像应该为不含噪声的图像，通过添加噪声作为训练样本。
[0024]在实际应用中也可以通过其他方法获取训练样本，但是必须保证各个训练样本对应的原始图像是已知。
[0025]训练样本的个数一方面影响到得到的训练好的神经网络的准确性，另一方面还会影响到训练的时间消耗，且二者之间相互制约。为保证二者之间的平衡，作为优选，所述网络训练样本集中包括8000?15000个训练样本(即获取8000?15000个大小与待处理图像相同原始图像)，进一步优选，所述网络训练样本集中包括10000个训练样本。
[0026]基于噪声独立同分布假设(即iid假设)对各个原始图像添加噪声。
[0027]根据噪声iid假设，对图像添加噪声构成训练集的输入部分，同时将对应的原始图像作为输出对网络进行监督。
[0028]针对当前原始图像I，通过如下公式添加噪声得到相应的训练样本X:
[0029]X = I+g
[0030]g ?gaussian ( δ )，
[0031]其中，g表示高斯噪声，均值为O ;
[0032]δ是高斯噪声的方差，其取值根据应用场景的噪声设定。
[0033]δ是高斯噪声的方差与应用场景的噪声大小相匹配，通常噪声越大，δ的取值应该越大。作为优选，所述S的取值为25?50。
[0034]所述步骤(3)中训练基本前馈神经网络包括如下步骤:
[0035](3-1)针对每个训练样本基于求解目标函数L的最小值构建相应的广义线性方程;
[0036]所述的目标函数L为:
[0037]L= (T-O)2，
[0038]其中，T表示目标输出，为当前训练样本对应的原始图像的向量表示，
[0039]O为当前训练样本在通过基本前馈神经网络的输出；
[0040](3-2)对将所有训练样本对应的广义线性方程组建为广义线性方程组，并使用Moore-Penrose逆求解所述的广义线性方程组得到特征抽取层到输出层的连接参数。
[0041]本发明中图像多采用矩阵形式表示，因此步骤(3-1)构建的广义线性方程实际上是一个小的广义线性方程组，而步骤(3-2)中将所有训练样本对应的广义线性方程组件得到的广义线性方程组，实际上应理解为若干个小的广义线性方程组构建的大广义线性方程组。
[0042]将经过添加噪声处理的图像(即训练样本)作为输入，对应的原始图像作为输出，求取网络中的参数，其中由输入到特征抽取层的参数随机初始化，不做改变，紧紧通过线性方程求取由特征抽取层到输出层的参数，大大节约了求解计算量，有利于降低训练时耗、缩短训练时间。
[0043]本方法通过构建三层网络，定制训练集，并通过ELM方式训练，可以很容易的获取网络参数，从而应用到对应的图像修复任务中，与现有技术相比，本发明具有如下有益效果:
[0044]通过定制训练集，并应用ELM方法对基本前馈神经网络进行训练获取连接参数，可以很迅速的完成训练过程，可以大大提高训练效率，缩短了训练时长；
[0045]根据应用场景的噪声类型构建网络训练集，仅需针对应用场景构建的网络训练集即可得到针对不同噪声类型的训练好的神经网络以对图像进行噪声去除，可以比较方便的应用到不同的噪声场景中；
[0046]在构建完成该网络时，可以针对不同噪声类型的图像完成定制，迅速完成训练，因此可以比较方便的应用到不同的噪声场景中；
[0047]训练完成后所需的内存和计算资源比较小，可以方便的应用在计算及存储资源受限场景中，如移动设备，嵌入式设备。
【附图说明】
[0048]图1为本实施例的基本前馈神经网络的示意图。
【具体实施方式】
[0049]下面将结合附图和具体实施例对本发明进行详细说明。
[0050]本实施例的基于ELM的图像去噪方法，包括如下步骤:
[0051](I)根据待处理图像的大小(记待处理图像的大小为MXN)构建基本网络。
[0052]如图1所示，本实施例的基本网络为基本前馈神经网络，包括三层，分别为输入层、特征抽取层和输出层。
[0053]输入层具有M*N个节点，输入待处理图像的像素信息；特征抽取层具有H个节点(H通常为1000?1500，本实施例中H = 1200);输出层具有M*N个节点，输出M*N大小的经过这种网络处理后的图像，即待处理图像去噪后的图像。
[0054]本实施例的前馈神经网络中:
[0055]从输入层到特征抽