本发明涉及图像处理技术领域,尤其涉及基于卷积导向图滤波的去噪方法。
背景技术:
图像去噪是非常重要的一个图像预处理环节,图像去噪是利用各种技术手段对噪声进行滤除工作,以提升图像的质量,使图像能够表达出原始图像的更多信息,为后续的图像处理工作提供一定的帮助。图像噪声来源广泛,不同的噪声类型会影响到噪声图像的恢复工作。因此,针对不同的噪声需要不同的去噪方法才能达到最优效果。
导向滤波是一种保边滤波,能够在平滑图像的同时保护物体边缘信息不丢失,并广泛应用于图像的降噪、边缘羽化、图像增强、抠图等图像处理中。与双边滤波相比,导向滤波对图像中物体边缘信息的保护力度更强,不会出现梯度反转的现象(表现为颜色异常的彩边),且时间复杂度更低。
目前,深度学习对图像去噪已经得到了广泛的应用,并且去噪效果表现优异,随着深度学习在计算机视觉领域中的飞速发展,越来越多的去噪算法基于神经网络进行研究实验,并得到了更优异的性能。深度学习技术因为其强大的学习能力,可通过大量数据训练实现端到端的自学习过程而十分适合应用于图像去噪。但是,目前的技术局限性在于,(1)用卷积神经网络进行图像去噪时,若网络层数太多,过于复杂,虽有较好的去噪效果,但是会导致边缘过于平滑;若网络结构简单,则会导致去噪不充分,部分区域会导致“蜡像感”严重。(2)导向图滤波可以达到保边滤波的效果,在平坦区域达到类似于均值滤波的去噪效果,从而较好的去除在平坦区域视频压缩造成的块噪声,但是对于边缘部分的毛刺噪声却无法去除。
本发明基于以往深度学习和导向滤波去噪的相关研究成果进行思考总结,提出泛化能力更强的去噪模型,从而获得更清晰真实的去噪结果。
技术实现要素:
由于现有方法存在上述问题,本发明实施例提出一种基于卷积导向图滤波的去噪方法。
具体地,本发明提供了以下技术方案,包括:
步骤11在高清图像中加入块噪声和毛刺噪声得到第一图像。
步骤12将所述第一图像输入cnn去噪网络得到第二图像。
步骤13根据所述第一图像和所述第二图像,对所述第一图像进行导向滤波得到第三图像。
具体地,以所述第二图像为导向图,第一图像为待滤波图,得到导向滤波系数,根据所述导向滤波系数,对第一图像进行导向滤波。
步骤14根据所述第三图像与所述高清图像计算loss,得到cnn去噪网络模型参数。
步骤15获取待去噪图像,输入所述已确定参数的cnn去噪网络进行去噪,得到第四图像。
步骤16根据所述待去噪图像和第四图像,对所述待去噪图像进行导向滤波,输出导向滤波去噪后的图像。
具体地,以所述待去噪图像作为待滤波图,所述第四图像为导向图,得到导向滤波系数。然后根据所述导向滤波系数,对所述待去噪图像进行导向滤波。
本发明的核心思想在于将卷积神经网络去噪和导向图滤波去噪的方法相结合,形成优势互补。由以上技术方案可知,本发明有以下有益效果:(1)将卷积神经网络结合导向滤波一起训练确定卷积神经网络参数使得卷积神经网络只需关注去除边缘区域的毛刺噪声,简化了卷积神经网络的复杂度;(2)结合卷积神经网络与导向滤波对待去噪图像进行去噪既能去除平坦区域的块噪声又能去除边缘区域的毛刺噪声,该去噪模型的去噪性能更加广泛。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图做简单的介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来说,在不付出创造性劳动的前提下,还可以根据这些图获得其他的附图。
图1表示本发明实施例基于卷积导向图滤波的去噪方法的流程示意图。
图2表示本发明实施例训练cnn模型的流程示意图。
图3表示本发明实施例训练cnn模型时通过下采样快速确定导向滤波系数的流程示意图。
图4表示本发明实施例基于卷积导向图滤波模型去噪过程中通过下采样快速确定导向滤波系数的流程示意图。
具体实施方式
下面结合附图,对本发明的具体实施方式作进一步描述。以下实施例仅用于更加清楚地说明本发明的技术方案,而不能以此来限制本发明的保护范围。
图1示出了本发明实施例提供的基于卷积导向图滤波的去噪方法的流程示意图,如图1所示,本发明实施例提供的基于卷积导向图滤波的去噪方法,具体包括如下内容:
步骤101在高清图像中加入块噪声和毛刺噪声,得到第一图像e。
可选地,所述第一图像可以是图片,也可以是视频中的图像帧。
步骤102根据所述第一图像e,得到cnn去噪网络去噪后的第二图像e'。
步骤103根据所述第一图像e以及所述cnn去噪网络去噪后的第二图像e',对第一图像e进行导向滤波。
具体地,作为一种实现方式,该步骤103具体可以包括:
以所述第一图像e作为待滤波图,所述cnn去噪后的第二图像e'作为导向图进行均值滤波,得到滤波后的图像meane和meane'
其中,r为滤波半径,设
计算近似的方差图
其中var表示方差,cov表示协方差,corr表示关系系数。
计算滤波系数k和m
其中,
根据所述导向滤波系数k和m对所述第一图像e进行导向滤波处理,输出第三图像e'',可以通过以下公式表示滤波过程:
可选地,为了降低计算复杂度,提高计算效率,作为另一种实现方式,如图3所示,步骤102-103可改进为:
将所述第一图像e作为第一分辨率图像进行下采样处理,得到第二分辨率图像esub。
将所述第二分辨率图像esub输入所述cnn去噪网络,输出cnn去噪网络去噪后的图像esub'。
根据所述第二分辨率图像esub和所述第二分辨率cnn去噪网络去噪后的图像esub',得到导向滤波系数ksub和msub。
具体地,所述得到导向图滤波系数ksub和msub,包括:
以所述第二分辨率图像esub作为待滤波图,所述第二分辨率cnn去噪网络去噪后的图像esub'作为导向图进行均值滤波,得到滤波后的图像
其中,r为滤波半径,设
计算近似的方差图
其中var表示方差,cov表示协方差,corr表示关系系数。
计算滤波系数ksub和msub
其中,
根据所述导向滤波系数ksub和msub进行与所述下采样处理相同的采样率进行上采样处理,得到所述第一分辨率图像的导向滤波系数k和m
其中s代表采样率,fup代表采样函数;
需要说明的是,下采样处理的采样率可以预先设定,本发明实施例不以此为限。
根据所述导向滤波系数k和m对所述第一图像e进行导向滤波处理,输出第三图像e'',可以通过以下公式表示滤波过程:
s104根据所述第三图像e''与所述高清图像计算loss,输出cnn去噪网络的模型参数。
本实施例中所述的第三图像e''在经过cnn去噪网络去噪后经过导向滤波处理,通过导向滤波处理去除了平坦区域的块噪声。以所述第三图像e''与所述高清图像计算loss,使得cnn去噪网络无需考虑平坦区域的噪声去除,只需要关注边缘的毛刺噪声的去除,极大地简化了cnn去噪网络的复杂程度,提高了去噪的效率,且平坦区域的噪声也能得到很好的去除。
s105获取待去噪图像p,输入所述已确定参数的cnn去噪网络,得到cnn去噪网络去噪后的第四图像p'。
可选地,所述待去噪图像p可以是图片也可以是视频中的图像帧。
s106根据所述待去噪图像p以及所述第四图像p',对所述待去噪图像p进行导向滤波,输出去噪后的图像o,可以通过以下公式表示滤波过程:
具体地,导向滤波系数u和v的计算方式与步骤103的计算方式相同,在此不做赘述。
可选地,为了降低计算复杂度,提高计算效率,导向滤波系数u和v的改进计算方式与前述步骤102-103的改进方式相同,在此不做赘述。
本发明的实施例中,将导向滤波与卷积神经网络结合,同时发挥了卷积神经网络和导向滤波的去噪优势,既去除了平坦区域的块噪声又出去了边缘区域的毛刺噪声。此外,还极大简化了卷积神经网络的层数和复杂度。可选地,在本发明的实施例中,结合上采样和下采样技术,极大降低了计算量,提高了计算速度。
通过以上的实施方式的描述,本领域的技术人员可以清楚地了解到各实施方式可借助软件加必需的通用硬件平台的方式来实现,当然也可以通过硬件。基于这样的理解,上述技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品可以存储在计算机可读存储介质中,如rom/ram、磁碟、光盘等,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行各个实施例或者实施例的某些部分所述的基于卷积导向图滤波的去噪方法。
此外,在本发明中,诸如“第一”、“第二”仅用于描述目的,而不能理解为指示或暗示相对重要性或者隐含指明所指示的技术特征的数量。由此,限定有“第一”、“第二”的特征可以明示或者隐含地包括至少一个该特征。
此外,在本发明中,诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来,而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。而且,术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含,从而使得包括一系列要素的过程、方法、物品或者设备不仅包括那些要素,而且还包括没有明确列出的其他要素,或者是还包括为这种过程、方法、物品或者设备所固有的要素。在没有更多限制的情况下,由语句“包括一个……”限定的要素,并不排除在包括所述要素的过程、方法、物品或者设备中还存在另外的相同要素。
最后应说明的是:以上实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的精神和范围。