排序卷积神经网络构建方法及其图像处理方法与装置与流程

文档序号:11831642阅读:334来源:国知局
排序卷积神经网络构建方法及其图像处理方法与装置与流程

本发明涉及图像处理技术,尤其涉及一种排序卷积神经网络构建方法及其图像处理方法与装置。



背景技术:

在图像处理领域,需要对一些呈现效果较差的图像进行处理,以获得呈现效果较佳的图像。例如,对于一幅雾霾图像,即在雾霾环境下所拍摄的图像,需要对其进行处理,以获得一幅清晰图像。

现有技术中,在进行雾霾图像处理时,首先通过手工计算的方式获取雾霾相关的特征,再通过手工计算获得的特征将雾霾图像恢复为清晰图像。

但是,使用手工计算来获取雾霾相关特征的方法,难以反映真实场景素材蕴含的客观规律,导致图像处理的效果差。



技术实现要素:

本发明供一种排序卷积神经网络构建方法及其图像处理方法与装置,用于解决使用现有技术导致的图像处理效果差的问题。

本发明第一方面提供一种排序卷积神经网络构建方法,所述排序卷积神经网络中包括:排序层、卷积层、下采样层以及全连接层,其中,所述排序层用于对所述排序层的前一层的输出结果进行排序处理,按照排序结果生成所述排序层的输出结果,并将所述排序层的输出结果输出到所述排序层的后一层,所述方法包括:

接收输入特征图,所述输入特征图使用二维矩阵表示;

按照所述排序层、所述卷积层、所述下采样层以及所述全连接层在所述排序卷积神经网络中的预设顺序,对所述输入特征图执行排序操作、卷积操作、下采样操作以及全连接操作,获取所述输入特征图对应的输出特征,所述输出特征为一个特征向量,其中,所述排序操作在所述排序层中进行,所述卷积操作在所述卷积层中进行,所述下采样操作在所述下采样层中进行,所述全连接操作在所述全连接层中进行;

将所述输出特征输出。

进一步地,所述排序操作在所述排序层中进行,包括:

接收所述排序层的前一层所输出的第一特征图,所述第一特征图使用二维矩阵表示;

对所述第一特征图中的所有元素进行排序;

按照排序结果,将所述第一特征图中的元素逐个写入第二特征图中;

将所述第二特征图输出到所述排序层的后一层。

进一步地,所述排序卷积神经网络包括十层,依次为:输入层、第一卷积层、第一下采样层、排序层、第二卷积层、第三卷积层、第二下采样层、第一全连接层、第二全连接层以及输出层,其中,所述第二卷积层的卷积核小于所述第一卷积层的卷积核,所述第三卷积层的卷积核小于所述第一卷积层的卷积核。

本发明第二方面提供一种图像处理方法,包括:

接收输入雾霾图像;

使用排序卷积神经网络确定所述输入雾霾图像的当前像素点的特征向量,其中,所述排序卷积神经网络中包括排序层、卷积层、下采样层以及全连接层,所述排序层用于对所述排序层的前一层的输出结果进行排序处理,按照排序结果生成所述排序层的输出结果,并将所述排序层的输出结果输出到所述排序层的后一层;

将所述当前像素点的特征向量输入随机森林回归模型中,获取所述当前像素点的雾霾浓度;

根据所述雾霾浓度以及环境光信息,确定所述输入雾霾图像对应的清晰图像。

进一步地,所述使用排序卷积神经网络确定所述输入雾霾图像的当前像素点的特征向量之前,还包括:

建立输出层特征向量与雾霾浓度之间的对应关系;

其中,所述输出层特征向量与雾霾浓度之间的对应关系为:对于输出层特征向量如果tΒ∈(i/10-0.1,i/10],则否则其中,tΒ为雾霾浓度。

进一步地,所述排序卷积神经网络包括十层,依次为:输入层、第一卷积层、第一下采样层、排序层、第二卷积层、第三卷积层、第二下采样层、第一全连接层、第二全连接层以及输出层,其中,所述第二卷积层的卷积核小于所述第一卷积层的卷积核,所述第三卷积层的卷积核小于所述第一卷积层的卷积核;

所述使用排序卷积神经网络确定所述输入雾霾图像的当前像素点的特征向量,包括:

通过所述输入层确定所述当前像素点的输入特征图,所述输入特征图使用二维矩阵表示;

对所述输入特征图进行排序处理、卷积处理、下采样以及全连接处理;

将所述第二全连接层输出的特征向量作为所述当前像素点的特征向量。

进一步地,所述将所述当前像素点的特征向量输入随机森林回归模型中,获取所述当前像素点的雾霾浓度,包括:

从所述当前像素点的特征向量中选取预设数量个元素训练所述随机森林回归模型中的回归树,获取所述随机森林回归模型的输出值,将所述随机森林回归模型的输出值作为所述当前像素点的雾霾浓度。

进一步地,所述根据所述雾霾浓度以及环境光信息,确定所述输入雾霾图像对应的清晰图像之后,还包括:

使用公式J*=λJ调整所述清晰图像的亮度,其中,Il(x)为输入雾霾图像的亮度,Jl(x)为清晰图像的亮度。

本发明第三方面提供一种排序卷积神经网络构建装置,所述排序卷积神经网络中包括:排序层、卷积层、下采样层以及全连接层,其中,所述排序层用于对所述排序层的前一层的输出结果进行排序处理,按照排序结果生成所述排序层的输出结果,并将所述排序层的输出结果输出到所述排序层的后一层,所述装置包括:

接收模块,用于接收输入特征图,所述输入特征图使用二维矩阵表示;

处理模块,用于按照所述排序层、所述卷积层、所述下采样层以及所述全连接层在所述排序卷积神经网络中的预设顺序,对所述输入特征图执行排序操作、卷积操作、下采样操作以及全连接操作,获取所述输入特征图对应的输出特征,所述输出特征为一个特征向量,其中,所述排序操作在所述排序层中进行,所述卷积操作在所述卷积层中进行,所述下采样操作在所述下采样层中进行,所述全连接操作在所述全连接层中进行;

输出模块,用于将所述输出特征输出。

进一步地,所述处理模块包括:

接收单元,用于接收排序层的前一层所输出的第一特征图,该第一特征图使用二维矩阵表示。

排序单元,用于对第一特征图中的所有元素进行排序。

写入单元,用于按照排序结果,将第一特征图中的元素逐个写入第二特征图中。

输出单元,用于将第二特征图输出到排序层的后一层。

进一步地,所述排序卷积神经网络包括十层,依次为:输入层、第一卷积层、第一下采样层、排序层、第二卷积层、第三卷积层、第二下采样层、第一全连接层、第二全连接层以及输出层,其中,所述第二卷积层的卷积核小于所述第一卷积层的卷积核,所述第三卷积层的卷积核小于所述第一卷积层的卷积核。

本发明第四方面提供一种图像处理装置,包括:

接收模块,用于接收输入雾霾图像;

第一确定模块,用于使用排序卷积神经网络确定所述输入雾霾图像的当前像素点的特征向量,其中,所述排序卷积神经网络中包括排序层、卷积层、下采样层以及全连接层,所述排序层用于对所述排序层的前一层的输出结果进行排序处理,按照排序结果生成所述排序层的输出结果,并将所述排序层的输出结果输出到所述排序层的后一层;

获取模块,用于将所述当前像素点的特征向量输入随机森林回归模型中,获取所述当前像素点的雾霾浓度;

计算模块,用于根据所述雾霾浓度以及环境光信息,计算所述输入雾霾图像对应的清晰图像。

进一步地,所述装置还包括:建立模块,用于建立输出层特征向量与雾霾浓度之间的对应关系;

其中,所述输出层特征向量与雾霾浓度之间的对应关系为:对于特征向量如果tΒ∈(i/10-0.1,i/10],则否则其中,tΒ为雾霾浓度。

进一步地,排序卷积神经网络包括十层,依次为:输入层、第一卷积层、第一下采样层、排序层、第二卷积层、第三卷积层、第二下采样层、第一全连接层、第二全连接层以及输出层,其中,第二卷积层的卷积核小于第一卷积层的卷积核,第三卷积层的卷积核小于第一卷积层的卷积核。所述第一确定模块包括:

确定单元,用于通过输入层确定当前像素点的输入特征图,该输入特征图使用二维矩阵表示。

处理单元,用于对输入特征图进行排序处理、卷积处理、下采样以及全连接处理。

输出单元,用于将第二全连接层输出的特征向量作为当前像素点的特征向量。

进一步地,所述获取模块具体用于:

从当前像素点的特征向量中选取预设数量个元素训练随机森林回归模型中的回归树,获取随机森林回归模型的输出值,将随机森林回归模型的输出值作为当前像素点的雾霾浓度。

进一步地,所述装置还包括:调整模块,用于使用公式J*=λJ调整所述清晰图像的亮度,其中,Il(x)为输入雾霾图像的亮度,Jl(x)为清晰图像的亮度。

本发明所提供的方法,通过构建排序卷积神经网络,该排序卷积神经网络能够通过自动学习来获取输入特征图对应的输出特征,相比与现有技术通过手工计算获取特征的方法,本发明的方法能够更好的反映真实场景素材蕴含的客观规律,将该方法应用于图像处理领域时,就能够使得图像处理的效果得到大幅提升。

附图说明

为了更清楚地说明本发明或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作一简单地介绍,显而易见地,下面描述中的附图是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动性的前提下,还可以根据这些附图获得其他的附图。

图1为本发明提供的排序卷积神经网络构建方法实施例一的流程示意图;

图2为本发明提供的排序卷积神经网络构建方法实施例二的流程示意图;

图3为排序层进行排序处理的示例图;

图4为本发明提供的图像处理方法实施例一的流程示意图;

图5为本发明提供的图像处理方法实施例二的流程示意图;

图6为本发明提供的排序卷积神经网络构建装置实施例一的模块结构图;

图7为本发明提供的排序卷积神经网络构建装置实施例二的模块结构图;

图8为本发明提供的图像处理装置实施例一的模块结构图;

图9为本发明提供的图像处理装置实施例二的模块结构图;

图10为本发明提供的图像处理装置实施例三的模块结构图;

图11为本发明提供的图像处理装置实施例四的模块结构图。

具体实施方式

为使本发明的目的、技术方案和优点更加清楚,下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。

图1为本发明提供的排序卷积神经网络构建方法实施例一的流程示意图,该方法中的排序卷积神经网络中包括:排序层、卷积层、下采样层以及全连接层,其中,排序层用于对排序层的前一层的输出结果进行排序处理,按照排序结果生成排序层的输出结果,并将排序层的输出结果输出到排序层的后一层。如图1所示,该方法包括:

S101、接收输入特征图,该输入特征图使用二维矩阵表示。

根据应用场景的不同,输入特征图中的元素所表示的含义不同,例如,当该方法应用于图像处理领域中是,输入特征图中的每个元素表示一个像素值。

S102、按照排序层、卷积层、下采样层以及全连接层在排序卷积神经网络中的预设顺序,对输入特征图执行排序操作、卷积操作、下采样操作以及全连接操作,获取输入特征图对应的输出特征,该输出特征为一个特征向量,其中,排序操作在排序层中进行,卷积操作在卷积层中进行,下采样操作在下采样层中进行,全连接操作在全连接层中进行。

排序卷积神经网络的架构可以根据实际需要进行灵活设置,例如,可以在排序卷积神经网络中设置1个或多个卷积层。

将卷积神经网络应用于不同领域时,统计特征都具有重要的作用,但是现有的卷积神经网络中对于统计特征的表达不足,因此,本发明在现有的卷积神经网络的基础上增加了排序层,形成新的排序卷积神经网络,排序层的数量可以根据需要进行灵活设置。增加了排序层的排序卷积神经网络能够增强神经网络对统计特征的描述能力,更好的学习出与所应用的领域相关的特征。

S103、将输出特征输出。

本实施例中,提出一种排序卷积神经网络构建方法,所构建的排序卷积神经网络能够通过自动学习来获取输入特征图对应的输出特征,相比于现有技术通过手工计算获取特征的方法,本发明的方法能够更好的反映真实场景素材蕴含的客观规律,将该方法应用于图像处理领域时,就能够使得图像处理的效果得到大幅提升。同时,本发明在卷积神经网络中增加了排序层,形成排序卷积神经网络,能够增强神经网络对统计特征的描述能力,更好的学习出与所应用的领域相关的特征。

在上述实施例的基础上,本实施例涉及排序层进行排序操作的具体方法,即,图2为本发明提供的排序卷积神经网络构建方法实施例二的流程示意图,如图2所示,排序层进行排序处理的具体方法为:

S201、接收排序层的前一层所输出的第一特征图,第一特征图使用二维矩阵表示。

例如,假设排序层的前一层为下采样层,则下采样层会向排序层输出一个特征图,即第一特征图,该特征图使用二维矩阵表示。

S202、对第一特征图中的所有元素进行排序。

具体地,可以根据实际需要对第一特征图中的元素从小到大进行排序,或者从大到小进行排序。

S203、按照排序结果,将第一特征图中的元素逐个写入第二特征图中。

具体地,第二特征图的大小与第一特征图相同,例如,假设第一特征图对应的二维矩阵为m×n,第一特征图对应的二维矩阵也为m×n。按照排序结果,将各元素按照行优先的顺序逐个写入第二特征图中,即将排序结果中排在第一位的元素写入第二特征图中的第一行第一列中,排在第二位的元素写入第二特征图中的第一行第二列中,依次类推。

S204、将第二特征图输出到排序层的后一层。

例如,假设排序层的后一层为卷积层,则会将第二特征图输出到卷积层,以供卷积层进行卷积处理。

图3为排序层进行排序处理的示例图,如图3所示,多个特征图经过排序层处理之后,其中的元素进行了重新排列,形成了新的特征图。

作为本发明的一种可选的实施方式,上述排序卷积神经网络中包括十层,依次为:输入层、第一卷积层、第一下采样层、排序层、第二卷积层、第三卷积层、第二下采样层、第一全连接层、第二全连接层以及输出层,其中,输出层也是一个全连接层。,第二卷积层的卷积核小于第一卷积层的卷积核,第三卷积层的卷积核小于第一卷积层的卷积核。以下以该十层排序卷积神经网络为例来说明上述步骤S102中排序卷积神经网络确定输出特征的过程,需要说明的而是,本示例中的数字仅是一种示例,不能作为对排序卷积神经网络的限制。

首先,收集400张清晰图像,并随机截取10万个20×20的清晰图像块。根据雾霾图像形成的物理过程及其数学描述,对于每个清晰图像块,假设环境光的RGB三个通道值均为1,随机选取10个雾霾浓度值,在清晰图像块上自动添加雾霾从而生成雾霾图像块,同时记录下每个雾霾图像块及其对应的雾霾浓度,共获得100万个训练图像块。

其次,输入层接收这100万个训练图像块,对于每个训练图像块,输入层会输出三个输入特征图,分别为训练图像块的R、G、B三个通道;第一卷积层接收输入特征图,并通过大小为5×5的卷积核进行卷积处理,输出32个16×16的特征图;第一下采样层采用2×2的采样窗口对32个16×16的特征图进行采样处理,输出32个8×8的特征图;排序层对32个8×8的特征图进行排序处,输出新的32个8×8的特征图;第二卷积层和第三卷积层的卷积核大小都为3×3,即比第一卷积层的卷积核小,通过第二卷积层和第三卷积层进行卷积处理之后,输出32个4×4的特征图;再经过第二下采样层使用2×2的采样窗口进行处理后,会输出32个2×2的特征图;第一全连接层和第二全连接层均为包含64维特征的全连接层,经过第一全连接层和第二全连接层的处理,会输出一个64维特征向量;最后,经过输出层处理后输出一个10维向量。

其中,第一卷积层、第二卷积层、第三卷积层以及第一全连接层使用了线性截断(Rectified linear unit,简称ReLU)激活函数,对于给定输入值x,ReLU激活函数的激活值为f(x)=max(x,0)。

本发明采用反向传播梯度下降法训练定义的排序卷积神经网络,并采用Softmax损失函数,损失函数定义为其中NΒ为训练样本的真实类别标签,YΒ为神经网络的预测结果,j对应的需要进一步说明的是,排序网络按照前向传播时对应关系反向传播梯度值。

再次,经过前面两个步骤之后,即完成了排序卷积神经网络的一次训练过程,输出层在本次训练结束时输出的特征向量的含义即为雾霾浓度,将此雾霾浓度与第一步时所记录的雾霾浓度值进行比较,根据差异调整排序卷积神经网络中各卷积层的卷积核的元素值以及各全连接层的中的特征矩阵,并使用新的卷积核和新的特征矩阵进行下一次的训练,依次类推,直到某次训练所输出的特征值与第一步时所记录的雾霾浓度值的差异在理想范围内,则完成了排序卷积神经网络的构建。

图4为本发明提供的图像处理方法实施例一的流程示意图,如图4所示,该方法包括:

S301、接收输入雾霾图像。

输入雾霾图像即为在雾霾环境下所拍摄的图像。

S302、使用排序卷积神经网络确定输入雾霾图像的当前像素点的特征向量。

其中,上述排序卷积神经网络中包括排序层、卷积层、下采样层以及全连接层,排序层用于对排序层的前一层的输出结果进行排序处理,按照排序结果生成排序层的输出结果,并将排序层的输出结果输出到排序层的后一层。

需要说明的是,对于一副输入雾霾图像中的每个像素点,都需要执行S302至S304的操作,即本发明是按照输入雾霾图像中的像素点逐个进行处理的。

S303、将当前像素点的特征向量输入随机森林回归模型中,获取当前像素点的雾霾浓度。

随机森林回归模型用于表示特征向量与雾霾浓度之间的映射关系,即对于特定的特征向量,通过在随即森林回归模型中进行训练,会得到与其对应的雾霾浓度。

优选地,在通过随机森林回归模型获取到雾霾浓度之后,需要通过引导滤波对雾霾浓度进行优化。

S304、根据雾霾浓度以及环境光信息,确定输入雾霾图像对应的清晰图像。

对于一副输入雾霾图像,每个像素点的雾霾浓度都可能不同,但是环境光在整副雾霾图像中都是相同的,因此容易估计得到。可选地,首先计算输入雾霾图像的暗黑通道,依据每个像素点的暗黑通道值,选取最亮的0.1%像素点,计算这些像素点的颜色平均值作为输入雾霾图像的环境光。

本实施例中,通过排序卷积神经网络来获取输入雾霾图像的特征向量,并通过随机森林回归模型确定特征向量对应的雾霾浓度,由于排序卷积神经网络能够自动学习出输入雾霾图像的特征,因此,能够更好的反映真实场景素材蕴含的客观规律,从而使得雾霾图像处理的效果得到大幅提升。

作为上述排序卷积神经网络的一种可选的实施方式,上述排序卷积神经网络包括十层,依次为:输入层、第一卷积层、第一下采样层、排序层、第二卷积层、第三卷积层、第二下采样层、第一全连接层、第二全连接层以及输出层,其中,输出层也是一个全连接层,其中,第二卷积层的卷积核小于第一卷积层的卷积核,第三卷积层的卷积核小于第一卷积层的卷积核。

图5为本发明提供的图像处理方法实施例二的流程示意图,如图5所示,基于该十层排序卷积神经网络来确定输入雾霾图像的特征向量的具体方法为:

S401、通过输入层确定所述像素点的输入特征图,该输入特征图使用二维矩阵表示。

当前像素点包括3个通道,对于3个通道会分别确定出一个输入特征图,即,本实施例所示的方法都是针对每个像素点的每个通道来逐个进行处理。

S402、对输入特征图进行排序处理、卷积处理、下采样以及全连接处理。

具体的方法可以参考前述排序卷积神经网络构建方法中的对应实施例,此处不再赘述。

S403、将第二全连接层输出的特征向量作为当前像素点的特征向量。

当训练排序卷积神经网络时,输出层输出的为一个特征向量,这个特征向量的含义即为雾霾浓度,但是,本发明还需要使用随机森林回归模型进行处理,如果直接将输出层所输出的特征向量输入到随机森林回归模型中,就不能有效地训练随机森林回归模型,因此,本发明构建一个十分类问题,将雾霾浓度值等分为10个区间,每个类别对应雾霾浓度值的一个区间,并通过训练排序卷积神经网络判别雾霾浓度属于哪个区间。即在训练排序卷积神经网络时,建立特征向量与雾霾浓度之间的对应关系,其中,特征向量与雾霾浓度之间的对应关系为:对于特征向量如果tΒ∈(i/10-0.1,i/10],则否则其中,tΒ为雾霾浓度。

建立特征向量和雾霾浓度之间的对应关系之后,在本实施例中,就将输出层的前一层,即第二全连接层所输出的特征向量,作为排序卷积神经网络的特征向量,并基于特征向量和雾霾浓度之间的对应关系对随机森林回归模型进行训练,从而得到特征向量对应的雾霾浓度值。

另一实施例中,上述步骤S303中将当前像素点的特征向量输入随机森林回归模型中,获取当前像素点的雾霾浓度的具体方法为:

从当前像素点的特征向量中选取预设数量个元素训练随机森林回归模型中的回归树,获取随机森林回归模型的输出值,将随机森林回归模型的输出值作为当前像素点的雾霾浓度。

例如,假设随机森林回归模型中包括200棵回归树,则从第二全连接层的64维向量中随机选取21维特征来训练每棵回归树,随机森林回归模型输出的即为雾霾浓度。

需要说明的是,在训练排序卷积神经网络时,所使用的训练样本具有R、G、B三个通道均为1的环境光,因此在使用排序卷积神经网络确定输入雾霾图像的特征向量之前,需要分别在每个通道上将输入雾霾图像除以环境光进行归一化,从而使得进行归一化之后的输入雾霾图像的环境光R、G、B三个通道均为1。

当使用前述方法得到输入雾霾图像的雾霾浓度和环境光之后,即可根据雾霾图像形成的物理过程及其数学公式I(x)=J(x)t(x)+A(1-t(x)),反向计算去除雾霾后的清晰图像其中,J(x)表示清晰图像,I(x)表示输入雾霾图像,t(x)表示雾霾浓度,A表示环境光。

得到的清晰图像J的亮度常常较低,因此,在得到清晰图像之后,还需要调整清晰图像的亮度,得到最终的清晰图像。

具体地,调整公式为J*=λJ,其中Il(x)为输入雾霾图像I(x)的亮度,Jl(x)为清晰图像J(x)的亮度。

图6为本发明提供的排序卷积神经网络构建装置实施例一的模块结构图,该排序卷积神经网络中包括:排序层、卷积层、下采样层以及全连接层,其中,排序层用于对排序层的前一层的输出结果进行排序处理,按照排序结果生成排序层的输出结果,并将排序层的输出结果输出到排序层的后一层。如图6所示,该装置包括:

接收模块501,用于接收输入特征图,该输入特征图使用二维矩阵表示。

处理模块502,用于按照排序层、卷积层、下采样层以及全连接层在排序卷积神经网络中的预设顺序,对输入特征图执行排序操作、卷积操作、下采样操作以及全连接操作,获取输入特征图对应的输出特征,该输出特征为特征向量,其中,排序操作在排序层中进行,卷积操作在卷积层中进行,下采样操作在下采样层中进行,全连接操作在全连接层中进行。

输出模块503,用于将输出特征输出。

该装置用于实现前述方法实施例,其实现原理和技术效果类似,此处不再赘述。

图7为本发明提供的排序卷积神经网络构建装置实施例二的模块结构图,如图7所示,处理模块502包括:

接收单元5021,用于接收排序层的前一层所输出的第一特征图,该第一特征图使用二维矩阵表示。

排序单元5022,用于对第一特征图中的所有元素进行排序。

写入单元5023,用于按照排序结果,将第一特征图中的元素逐个写入第二特征图中。

输出单元5024,用于将第二特征图输出到排序层的后一层。

另一实施例中,所述排序卷积神经网络包括十层,依次为:输入层、第一卷积层、第一下采样层、排序层、第二卷积层、第三卷积层、第二下采样层、第一全连接层、第二全连接层以及输出层,其中,所述第二卷积层的卷积核小于所述第一卷积层的卷积核,所述第三卷积层的卷积核小于所述第一卷积层的卷积核。

图8为本发明提供的图像处理装置实施例一的模块结构图,如图8所示,该装置包括:

接收模块601,用于接收输入雾霾图像。

第一确定模块602,用于使用排序卷积神经网络确定输入雾霾图像的当前像素点的特征向量,其中,排序卷积神经网络中包括排序层、卷积层、下采样层以及全连接层,排序层用于对排序层的前一层的输出结果进行排序处理,按照排序结果生成排序层的输出结果,并将排序层的输出结果输出到排序层的后一层。

获取模块603,用于将当前像素点的特征向量输入随机森林回归模型中,获取当前像素点的雾霾浓度。

计算模块604,用于根据雾霾浓度以及环境光信息,计算输入雾霾图像对应的清晰图像。

该装置用于实现前述方法实施例,其实现原理和技术效果类似,此处不再赘述。

图9为本发明提供的图像处理装置实施例二的模块结构图,如图9所示,在图8的基础上,该装置还包括:

建立模块605,用于建立所述特征向量与雾霾浓度之间的对应关系;

其中,所述特征向量与雾霾浓度之间的对应关系为:对于特征向量如果tΒ∈(i/10-0.1,i/10],则否则其中,tΒ为雾霾浓度。

另一实施例中,排序卷积神经网络包括十层,依次为:输入层、第一卷积层、第一下采样层、排序层、第二卷积层、第三卷积层、第二下采样层、第一全连接层、第二全连接层以及输出层,其中,输出层也是一个全连接层,其中,第二卷积层的卷积核小于第一卷积层的卷积核,第三卷积层的卷积核小于第一卷积层的卷积核。

图10为本发明提供的图像处理装置实施例三的模块结构图,如图10所示,基于上述十层排序卷积神经网络,第一确定模块602包括:

确定单元6021,用于通过输入层确定当前像素点的输入特征图,该输入特征图使用二维矩阵表示。

处理单元6022,用于对输入特征图进行排序处理、卷积处理、下采样以及全连接处理。

输出单元6023,用于将第二全连接层输出的特征向量作为当前像素点的

特征向量。

另一实施例中,获取模块603具体用于:

从当前像素点的特征向量中选取预设数量个元素训练随机森林回归模型中的回归树,获取随机森林回归模型的输出值,将随机森林回归模型的输出值作为当前像素点的雾霾浓度。

图11为本发明提供的图像处理装置实施例四的模块结构图,如图11所示,在图9的基础上,该装置还包括:

调整模块606,用于使用公式J*=λJ调整所述清晰图像的亮度,其中,Il(x)为输入雾霾图像的亮度,Jl(x)为清晰图像的亮度。

本领域普通技术人员可以理解:实现上述各方法实施例的全部或部分步骤可以通过程序指令相关的硬件来完成。前述的程序可以存储于一计算机可读取存储介质中。该程序在执行时,执行包括上述各方法实施例的步骤;而前述的存储介质包括:ROM、RAM、磁碟或者光盘等各种可以存储程序代码的介质。

最后应说明的是:以上各实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述各实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分或者全部技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的范围。

当前第1页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1