CN116433525A - 一种基于边缘检测函数变分模型的水下图像去雾方法 - Google Patents

一种基于边缘检测函数变分模型的水下图像去雾方法 Download PDF

Info

Publication number
CN116433525A
CN116433525A CN202310412291.4A CN202310412291A CN116433525A CN 116433525 A CN116433525 A CN 116433525A CN 202310412291 A CN202310412291 A CN 202310412291A CN 116433525 A CN116433525 A CN 116433525A
Authority
CN
China
Prior art keywords
underwater
image
defogging
edge detection
following
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Pending
Application number
CN202310412291.4A
Other languages
English (en)
Inventor
杨婧
金正猛
唐维
巢腾
马悦
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Nanjing University of Posts and Telecommunications
Original Assignee
Nanjing University of Posts and Telecommunications
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Nanjing University of Posts and Telecommunications filed Critical Nanjing University of Posts and Telecommunications
Priority to CN202310412291.4A priority Critical patent/CN116433525A/zh
Publication of CN116433525A publication Critical patent/CN116433525A/zh
Pending legal-status Critical Current

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T5/00Image enhancement or restoration
    • G06T5/73Deblurring; Sharpening
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/50Depth or shape recovery
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/10Image acquisition modality
    • G06T2207/10024Color image

Landscapes

  • Engineering & Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Theoretical Computer Science (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Image Processing (AREA)

Abstract

本发明涉及计算机视觉领域中的图像处理领域,具体地说,是一种基于边缘检测函数变分模型的水下图像去雾方法,其中,提出了一种具有边缘检测函数的水下暗通道先验来估计初始传输,确保可以获得具有更多细节的相应初始场景深度,通过初始场景深度,可以很好地保留去雾图像的一些细节特征;利用场景深度的分段平滑先验和无雾图像的分段常数先验,提出了一种新的具有TGV和TV正则化的变分模型,用于同时对水下图像进行去雾和去噪;建立了该模型的极小值的存在性和唯一性,还设计了一种快速算法来数值求解该模型,所设计的算法具有收敛性,实验结果表明了该发明的有效性。

Description

一种基于边缘检测函数变分模型的水下图像去雾方法
技术领域
本发明涉及计算机视觉领域中的图像处理领域,具体地说,是一种基于边缘检测函数变分模型的水下图像去雾方法。
背景技术
水下世界蕴藏着丰富的资源。近年来,水下成像的研究和应用越来越受到人们的关注,但由于水下光受散射、吸收和噪声的影响,获取的水下图像通常存在雾度、低对比度、清晰度差等问题。现有的水下图像去雾方法通常会导致去雾图像中的颜色失真和噪声放大。为了提高水下图像的可视性,已经提出了极化成像和立体成像方法,以获得高质量的水下图像。然而,这些基于硬件的方法是昂贵的,并且在实际的应用中受到限制。与高成本的硬件方法相比,低成本的数字图像处理软件方法具有更大的实用价值。
基于先验知识的方法是基于软件的方法的代表。众所周知,传输图的估计在获得无雾的水下图像中起着至关重要的作用。最初提出了暗通道先验(Dark Channel Prior,DCP)来估计传输图。随后,对DCP衍生的水下图像去雾又进行了许多研究。由于红光的最大衰减,传统的DCP并不完全适用于水下环境,所以仅考虑蓝色和绿色通道的视觉信息,由此引入了水下暗通道先验(Underwater Dark Channel Prior,UDCP)。UDCP的引入使基于DCP的现有方法得到了显著改进。之后,提出了各种DCP的先验知识和变体,用于传输估计和水下图像恢复。例如:提出了一种红色通道先验(RedChannel Prior,RCP)来恢复蓝绿色图像,恢复了与短波长相关的颜色;利用最大强度先验(Maximum IntensityPrior,MIP)估计背景光,提出了一种基于蓝绿通道去阴影和红通道校正的水下图像恢复方法;基于距离相机较远的物体在水下成像时更模糊的事实,采用图像模糊先验(Blurriness Prior,BP)来提高估计透射和场景深度的准确性。能发现:大多数水下图像恢复方法只关注如何改进传输图的估计,而忽略了水下图像噪声的影响。
为了克服水下图像噪声放大和颜色失真的问题,近几十年来提出了一些水下图像去雾和去噪同时进行的变分方法。例如,水下暗信道先验被应用于估计传输图,并提出了一种具有非局部全变差(Total Variation,TV)图像去雾模型用于水下图像去雾去噪。这种非局部TV方法可以在去雾和去噪方面获得令人满意的性能,但由于应用了非局部微分算子,因此耗时较长。最近,又建立了一种新颖的基于TV和曲率的方法来恢复水下图像。值得一提的是,卷积神经网络(Convolutional Neural Network,CNN)已被应用于图像去雾。例如,提出了一种称为DehazeNet的可练端到端网络,用于从模糊图像中直接学习和估计传输图。此外,基于CNN的网络UIE Net被提出用于水下图像增强。
大部分算法只能实现水下图像色彩校正或图像对比度改善,对于水下图像存在的亮暗分布不均匀校正效果较差;同时没有充分考虑到水下光学成像物理属性,往往容易导致过增强的现象,具体表现为增强后的图像色彩过饱和或颜色失真。
发明内容
为了解决上述技术问题,本发明披露了一种基于边缘检测函数变分模型的水下图像去雾方法,具体包括以下步骤:
步骤1、引入图像去雾的必要符号、水下暗通道先验及非局部TV模型,并提出一种具有边缘检测函数的水下暗通道先验如下:
Figure BDA0004183612640000021
此处的
Figure BDA0004183612640000022
是边缘检测函数,其中σ>0,H是I的灰度图像;
水下有雾原像I记为:
I:Ω→R3={(IR,IG,IB):IR,IG,IB>0}
可以得到它的灰度图像:
H=0.2989*IR+0.5870*IG+0.1140*IB
有如下的水下光学模型:
Figure BDA0004183612640000023
其中J=(JR,JG,JB)是还原后的图像,背景光K=(KR,KG,KB)考虑为全局常量,透射率t可以重新表成指数衰减形式:
Figure BDA0004183612640000031
这里d为景深函数,表示x点物体到摄像机的距离,η>0一般设置为1,设g=log(K-J),f=log(K-I)和d=-logt,将水下光学模型转化为:
g=f+d,
其中,g=(gR,gG,gB),f=(fR,fG,fB)与d=(d,d,d)无雾图像即为J=K-exp(g),水下暗通道先验(UDCP)是一个处理水下图像去雾问题的简单方法:
Figure BDA0004183612640000032
其中,Ω为以x为中心的网格,认为JUDCP=0,可以得到透射率表达式:
Figure BDA0004183612640000033
可得景深d0=logt0,可以计算出g和J;
引入了如下的水下去雾非局部TV(Nonlocal Total Variation,NLTV)模型:
Figure BDA0004183612640000034
其中,
Figure BDA0004183612640000035
是非局部微分算子,α和β是正参数;
在灰度图H的边缘,边缘检测函数ψ趋向于0,而在图像的平滑区域它趋向于1,考虑JEI-UDCP=0,根据EI-UDCP先验,得到去雾图像的景深d0=-logt0,其中透射率t0可表为:
Figure BDA0004183612640000041
根据式(1)有:
Figure BDA0004183612640000042
利用JEI-UDCP=0,易得透射率t0表达式。
步骤2、建立变分模型;
考虑如下的变分问题:
Figure BDA0004183612640000043
其中
Figure BDA0004183612640000044
α1,γ及λ=(λ01)为正参数,BV(Ω;R3)和
Figure BDA0004183612640000045
分别代表向量值有界全变差(bounded variation)空间和二阶有界广义全变差(bounded generalized variationwith second order)空间,离散形式的/>
Figure BDA00041836126400000410
可表为如下的最小值形式:
Figure BDA0004183612640000046
最小值在Ω上的所有实向量场ω中取,并且
Figure BDA0004183612640000047
代表对称导,研究变分模型的等价形式:
Figure BDA0004183612640000048
Figure BDA0004183612640000049
步骤3、建立变分模型的最小值的存在性和唯一性;
定义1:定义BV(Ω,R3)为包含满足下面条件的g的函数空间:
(1)g∈L1(Ω;R3);
(2)
Figure BDA0004183612640000051
取有限值,BV(Ω;R3)是一个Banach空间,有范数:/>
Figure BDA0004183612640000052
定义2:令
Figure BDA0004183612640000053
为有界域且(λ01),其中λ01均为正数。对于/>
Figure BDA0004183612640000054
带权值λ的二阶广义全变差(TGVofordertwo)定义如下:
Figure BDA0004183612640000055
其中,
Figure BDA0004183612640000056
为仅在Ω上,连续可微的对称矩阵域空间,Sn×n表示所有n×n的对称矩阵,空间/>
Figure BDA0004183612640000057
有范数:
Figure BDA0004183612640000058
称作为向量值函数的二阶有界广义全变差空间(BGV oforder two);
引理1:假设gk∈BV(Ω;R3)(k=1,2,..),并且在
Figure BDA0004183612640000059
那么
Figure BDA00041836126400000510
引理2:假设dk∈BGV(Ω;R3)(k=1,2,..),并且在
Figure BDA00041836126400000511
上dk→d,那么
Figure BDA00041836126400000512
引理3:假设
Figure BDA00041836126400000513
是一列定义在BV(Ω;R3)上,满足/>
Figure BDA00041836126400000514
于是存在一个收敛子列/>
Figure BDA00041836126400000515
和一个函数g∈BV(Ω;R3),使得/>
Figure BDA00041836126400000516
引理4:假设
Figure BDA00041836126400000517
是一列定义在BGV(Ω;R3)上,满足/>
Figure BDA00041836126400000518
于是存在一个收敛子列/>
Figure BDA00041836126400000519
和一个函数d∈BGV(Ω;R3),使得/>
Figure BDA0004183612640000061
把变分问题的能量泛函写为E(g,d);
定理1:令f∈L2(Ω;R3)且d0∈L2(Ω;R3),于是对于固定的参数α1,γ>0,能量泛函有唯一的最小化方法。
步骤4、记输入图像的规模为M×N,令X=RMN为有限维实向量空间,定义了内积<·,·>X
Figure BDA0004183612640000062
其中梯度/>
Figure BDA0004183612640000063
是向量空间Y=X×X上的向量;
Figure BDA0004183612640000064
算子符合Neumann边界条件的离散形式定义为:
Figure BDA0004183612640000065
其中,
Figure BDA0004183612640000066
定义:
P1={p1∈R6MN:p1≤α1},P2={p2∈R6MN:p2≤λ1},Q={q∈R12MN:q≤λ0},
并且记闭凸集C上的指示函数IC为:
Figure BDA0004183612640000067
对于一个真凸并且下半连续的函数Φ,定义近端算子
Figure BDA00041836126400000610
为:
Figure BDA0004183612640000068
于是变分问题等价形式可以重新写作下面的鞍点问题:
Figure BDA0004183612640000069
其中p1,p2,q是对偶变量;
通过原始对偶算法,任给一个第k步的中间结果
Figure BDA0004183612640000071
qk,gk,dkk,更新解的过程如下:
③更新变量
Figure BDA0004183612640000072
给定gk,dkk,更新/>
Figure BDA0004183612640000073
为:
Figure BDA0004183612640000074
Figure BDA0004183612640000075
Figure BDA0004183612640000076
其中
Figure BDA0004183612640000077
④更新变量dk+1,gk+1k+1.记对称导的伴随算子为:
Figure BDA0004183612640000078
其中
Figure BDA0004183612640000079
给定/>
Figure BDA00041836126400000710
可以更新更新变量dk+1,gk+1k+1为:
Figure BDA0004183612640000081
Figure BDA0004183612640000082
Figure BDA0004183612640000083
其中,
Figure BDA0004183612640000084
进一步,容易得到:
Figure BDA0004183612640000085
最终得到了水下去雾算法流程如下:
重复迭代:更新
Figure BDA0004183612640000086
如下:
Figure BDA0004183612640000087
Figure BDA0004183612640000088
Figure BDA0004183612640000089
Figure BDA00041836126400000810
Figure BDA00041836126400000811
Figure BDA0004183612640000091
Figure BDA0004183612640000092
Figure BDA0004183612640000093
Figure BDA0004183612640000094
迭代终止条件:
Figure BDA0004183612640000095
定理2:记
Figure BDA0004183612640000096
假定正参数σ1212满足/>
Figure BDA0004183612640000097
和/>
Figure BDA0004183612640000098
于是得到算法产生的序列/>
Figure BDA0004183612640000099
收敛到/>
Figure BDA00041836126400000910
也就是鞍点问题的解。
步骤5、对步骤2中的模型利用MATLAB进行实验,得到去雾图像,用峰值信噪比(PSNR)、结构相似度(SSIM)评价不同方法对合成水下有雾图像去雾的表现。用水下有色图像质量评估(UCIQE)、水下图像质量测度(UIQM)、水下图像清晰度测度(UISM)和图像块对比度质量指数(PCQI)来评价不同方法对真实水下有雾图像的去雾效果。
对于这些准则来说,更大的数值代表了更好的去雾结果。
评价指标峰值信噪比(PSNR)为:
Figure BDA00041836126400000911
结构相似性(SSIM)为:
Figure BDA00041836126400000912
其中,u0和u分别表示原图像和去雾图像,
Figure BDA00041836126400000913
和μu是相应的平均值,/>
Figure BDA00041836126400000914
和σu代表它们的标准离差,/>
Figure BDA00041836126400000915
是u0和u的协方差,此外c1,c2是常量。
本发明的有益效果:本发明提出了一种具有边缘检测函数的水下暗通道先验(EI-UDCP)来估计初始传输,确保可以获得具有更多细节的相应初始场景深度,通过初始场景深度,可以很好地保留去雾图像的一些细节特征;利用场景深度的分段平滑先验和无雾图像的分段常数先验,提出了一种新的具有TGV和TV正则化的变分模型,用于同时对水下图像进行去雾和去噪;建立了该模型的极小值的存在性和唯一性,还设计了一种快速算法来数值求解该模型,所设计的算法具有收敛性,实验结果表明了该发明的有效性。
附图说明
图1为本发明的流程示意图。
具体实施方式
为了加深对本发明的理解,下面将结合实施例对本发明做进一步详细描述,该实施例仅用于解释本发明,并不对本发明的保护范围构成限定。
实施例:一种基于边缘检测函数变分模型的水下图像去雾方法,包括如下步骤:
步骤1、引入图像去雾的必要符号、水下暗通道先验及非局部TV模型,并提出一种具有边缘检测函数的水下暗通道先验如下:
Figure BDA0004183612640000101
此处的
Figure BDA0004183612640000102
是边缘检测函数,其中σ>0,H是I的灰度图像;
水下有雾原像I记为:
I:Ω→R3={(IR,IG,IB):IR,IG,IB>0}
可以得到它的灰度图像:
H=0.2989*IR+0.5870*IG+0.1140*IB
有如下的水下光学模型:
Figure BDA0004183612640000103
其中,J=(JR,JG,JB)是还原后的图像,背景光K=(KR,KG,KB)考虑为全局常量,透射率t可以重新表成指数衰减形式:
Figure BDA0004183612640000111
这里d为景深函数,表示x点物体到摄像机的距离,η>0一般设置为1,设g=log(K-J),f=log(K-I)和d=-logt,将水下光学模型转化为:
g=f+d,
其中,g=(gR,gG,gB),f=(fR,fG,fB)与d=(d,d,d)无雾图像即为J=K-exp(g),水下暗通道先验(UDCP)是一个处理水下图像去雾问题的简单方法:
Figure BDA0004183612640000112
其中,Ω为以x为中心的网格,认为JUDCP=0,可以得到透射率表达式:
Figure BDA0004183612640000113
可得景深d0=logt0,可以计算出g和J;
引入了如下的水下去雾非局部TV(Nonlocal Total Variation,NLTV)模型:
Figure BDA0004183612640000114
其中
Figure BDA0004183612640000115
是非局部微分算子,α和β是正参数;
在灰度图H的边缘,边缘检测函数ψ趋向于0,而在图像的平滑区域它趋向于1,考虑JEI-UDCP=0,根据EI-UDCP先验,得到去雾图像的景深d0=-logt0,其中透射率t0可表为:
Figure BDA0004183612640000121
根据式(1)有:
Figure BDA0004183612640000122
利用JEI-UDCP=0,易得透射率t0表达式。
步骤2、建立变分模型。
考虑如下的变分问题:
Figure BDA0004183612640000123
其中
Figure BDA0004183612640000124
α1,γ及λ=(λ01)为正参数,BV(Ω;R3)和
Figure BDA0004183612640000125
分别代表向量值有界全变差(bounded variation)空间和二阶有界广义全变差(bounded generalized variationwith second order)空间,离散形式的/>
Figure BDA0004183612640000129
可表为如下的最小值形式:
Figure BDA0004183612640000126
最小值在Ω上的所有实向量场ω中取,并且
Figure BDA0004183612640000127
代表对称导,研究变分模型的等价形式:
Figure BDA0004183612640000128
步骤3、建立变分模型的最小值的存在性和唯一性。
定义1:定义BV(Ω,R3)为包含满足下面条件的g的函数空间:
(1)g∈L1(Ω;R3);
(2)
Figure BDA0004183612640000131
取有限值。BV(Ω;R3)是一个Banach空间,有范数:/>
Figure BDA0004183612640000132
定义2:令
Figure BDA00041836126400001318
为有界域且(λ01),其中λ01均为正数。
对于
Figure BDA0004183612640000133
带权值λ的二阶广义全变差(TGVofordertwo)定义如下:
Figure BDA0004183612640000134
其中,
Figure BDA0004183612640000135
为仅在Ω上,连续可微的对称矩阵域空间,Sn×n表示所有n×n的对称矩阵,空间/>
Figure BDA0004183612640000136
有范数:
Figure BDA0004183612640000137
称作为向量值函数的二阶有界广义全变差空间(BGVoforder two);
引理1:假设gk∈BV(Ω;R3)(k=1,2,..),并且在
Figure BDA0004183612640000138
那么
Figure BDA0004183612640000139
引理:2:假设dk∈BGV(Ω;R3)(k=1,2,..),并且在
Figure BDA00041836126400001310
上dk→d,那么
Figure BDA00041836126400001311
引理3:假设
Figure BDA00041836126400001312
是一列定义在BV(Ω;R3)上,满足/>
Figure BDA00041836126400001313
于是存在一个收敛子列/>
Figure BDA00041836126400001314
和一个函数g∈BV(Ω;R3),使得/>
Figure BDA00041836126400001315
引理4:假设
Figure BDA00041836126400001316
是一列定义在BGV(Ω;R3)上,满足/>
Figure BDA00041836126400001317
于是存在一个收敛子列/>
Figure BDA0004183612640000141
和一个函数d∈BGV(Ω;R3),使得/>
Figure BDA0004183612640000142
把变分问题的能量泛函写为E(g,d);
定理1:令f∈L2(Ω;R3)且d0∈L2(Ω;R3),于是对于固定的参数α1,γ>0,能量泛函有唯一的最小化方法。
步骤4、记输入图像的规模为M×N,令X=RMN为有限维实向量空间,定义了内积<·,·>X
Figure BDA0004183612640000143
其中梯度/>
Figure BDA0004183612640000144
是向量空间Y=X×X上的向量;
Figure BDA0004183612640000145
算子符合Neumann边界条件的离散形式定义为:
Figure BDA0004183612640000146
其中,
Figure BDA0004183612640000147
定义
P1={p1∈R6MN:p1≤α1},P2={p2∈R6MN:p2≤λ1},Q={q∈R12MN:q≤λ0},
并且记闭凸集C上的指示函数IC为:
Figure BDA0004183612640000148
对于一个真凸并且下半连续的函数Φ,定义近端算子
Figure BDA0004183612640000149
为:
Figure BDA00041836126400001410
于是变分问题等价形式可以重新写作下面的鞍点问题:
Figure BDA0004183612640000151
其中p1,p2,q是对偶变量;
通过原始对偶算法,任给一个第k步的中间结果
Figure BDA0004183612640000152
qk,gk,dkk,更新解的过程如下:
⑤更新变量
Figure BDA0004183612640000153
给定gk,dkk,更新/>
Figure BDA0004183612640000154
为:
Figure BDA0004183612640000155
Figure BDA0004183612640000156
Figure BDA0004183612640000157
其中
Figure BDA0004183612640000158
⑥更新变量dk+1,gk+1k+1.记对称导的伴随算子为:
Figure BDA0004183612640000159
其中
Figure BDA0004183612640000161
给定/>
Figure BDA0004183612640000162
可以更新更新变量dk+1,gk+1k+1为:
Figure BDA0004183612640000163
/>
Figure BDA0004183612640000164
Figure BDA0004183612640000165
其中
Figure BDA0004183612640000166
进一步,容易得到:
Figure BDA0004183612640000167
最终得到了水下去雾算法流程如下:
重复迭代:更新
Figure BDA0004183612640000168
如下:
Figure BDA0004183612640000169
Figure BDA00041836126400001610
Figure BDA00041836126400001611
Figure BDA00041836126400001612
Figure BDA0004183612640000171
Figure BDA0004183612640000172
Figure BDA0004183612640000173
Figure BDA0004183612640000174
Figure BDA0004183612640000175
迭代终止条件:
Figure BDA0004183612640000176
定理2:记
Figure BDA0004183612640000177
假定正参数σ1212满足/>
Figure BDA0004183612640000178
和/>
Figure BDA0004183612640000179
于是得到,所提出的算法产生的序列/>
Figure BDA00041836126400001710
收敛到/>
Figure BDA00041836126400001711
也就是鞍点问题的解。
步骤5、对步骤2中的模型利用MATLAB进行实验,得到去雾图像,用峰值信噪比(PSNR)、结构相似度(SSIM)评价不同方法对合成水下有雾图像去雾的表现。用水下有色图像质量评估(UCIQE)、水下图像质量测度(UIQM)、水下图像清晰度测度(UISM)和图像块对比度质量指数(PCQI)来评价不同方法对真实水下有雾图像的去雾效果。对于这些准则来说,更大的数值代表了更好的去雾结果。
评价指标峰值信噪比(PSNR)为:
Figure BDA00041836126400001712
结构相似性(SSIM)为:
Figure BDA00041836126400001713
其中,u0和u分别表示原图像和去雾图像,
Figure BDA00041836126400001714
和μu是相应的平均值,/>
Figure BDA00041836126400001715
和σu代表它们的标准离差,/>
Figure BDA00041836126400001716
是u0和u的协方差,此外c1、c2是常量。
对于本发明的所有测试,采用一个11×11的窗口来获得每个像素位置x∈Ω。
在所提出的模型中,有四个参数α1、λ1、λ0和γ。在所有测试中令λ1=0.06、λ0=1、γ=4。
根据输入的水下图像的噪声水平调整参数α1,当输入图像的噪声水平较高时,其值会较大;根据输入图像的噪声水平将α1设置为0.02、0.06或0.1。对于所有测试图像,α1的详细值列于表1。
表1所提出模型的α1的值
Figure BDA0004183612640000181
在算法中,选择
Figure BDA0004183612640000182
停止条件设置为/>
Figure BDA0004183612640000183
使用MATLAB 2019b在***为Windows7的计算机上进行实验,CPU的具体型号为Intel(R)Core(TM)i5-6200U [email protected]
表2中粗体数字表示每行中的最佳值,可以看出本发明的平均PSNR和SSIM值高于其他方法。这些结果表明,本发明可以很好地去除合成的水下图像的雾,获得更好的视觉效果。
表2去雾结果PSNR和SSIM值
Figure BDA0004183612640000184
Figure BDA0004183612640000191
表3中粗体数字表示每行中的最佳值,可以看出最高的平均UCIQE、UIQM和PCQI值显示本发明在饱和度和对比度之间达成了很好的平衡。值得一提的是,本发明有着最高的平均UISM值,这表明EI-UDCP方法确实可以很好地保留水下图片的边缘信息。
表3去雾结果的UCIQE,UIQM,PCQI和UISM值
Figure BDA0004183612640000192
Figure BDA0004183612640000201
通过表4可以看出,Galdran和Peng的方法在时间上的消耗较短,因为它们产生的图片直接基于水下光学模型。DehazeNet使用了训练后的模型参数来进行水下图像的去雾,因此也相对较快。不过,相对于NLTV模型来说,本发明更快更高效。
表4对比平均计算时间(单位:秒)
Figure BDA0004183612640000202
表5中粗体数字表示每列中的最佳值,计算出PSNR和SSIM值用以量化评价去雾和降噪的效果。可以看出本发明总体而言有着更好的数值。本发明在带噪图像去雾时依然有着优秀的结果。
表5PSNR和SSIM值
Figure BDA0004183612640000203
以上显示和描述了本发明的基本原理、主要特征及优点。本行业的技术人员应该了解,本发明不受上述实施例的限制,上述实施例和说明书中描述的只是说明本发明的原理,在不脱离本发明精神和范围的前提下,本发明还会有各种变化和改进,这些变化和改进都落入要求保护的本发明范围内。本发明要求保护范围由所附的权利要求书及其等效物界定。

Claims (7)

1.一种基于边缘检测函数变分模型的水下图像去雾方法,其特征在于,包括以下步骤:
步骤1、引入图像去雾的必要符号、水下暗通道先验及非局部TV模型,并提出一种具有边缘检测函数的水下暗通道先验;
步骤2、建立变分模型;
步骤3、建立变分模型的最小值的存在性和唯一性;
步骤4、定义待检测水下图像;
步骤5、对步骤2中的模型利用MATLAB进行实验,得到去雾图像,用峰值信噪比、结构相似度评价不同方法对合成水下有雾图像去雾的表现。
2.根据权利要求1所述的基于边缘检测函数变分模型的水下图像去雾方法,其特征在于,所述步骤1中,一种具有边缘检测函数的水下暗通道先验如下:
Figure QLYQS_1
其中,
Figure QLYQS_2
是边缘检测函数,其中σ>0,H是I的灰度图像;
水下有雾原像I记为:
I:Ω→R3={(IR,IG,IB):IR,IG,IB>0}
得到它的灰度图像:
H=0.2989*IR+0.5870*IG+0.1140*IB
有如下的水下光学模型:
Figure QLYQS_3
其中J=(JR,JG,JB)是还原后的图像,背景光K=(KR,KG,KB)考虑为全局常量,透射率t可以重新表成指数衰减形式:
Figure QLYQS_4
其中,d为景深函数,表示x点物体到摄像机的距离,η>0一般设置为1,设g=log(K-J),f=log(K-I)和d=-logt,将水下光学模型转化为:
g=f+d,
其中,g=(gR,gG,gB),f=(fR,fG,fB)与d=(d,d,d)无雾图像即为J=K-exp(g),水下暗通道先验是一个处理水下图像去雾问题的简单方法:
Figure QLYQS_5
其中,Ω为以x为中心的网格,认为JUDCP=0,得到透射率表达式:
Figure QLYQS_6
可得景深d0=logt0,可以计算出g和J;
引入了如下的水下去雾非局部TV模型:
Figure QLYQS_7
其中,
Figure QLYQS_8
是非局部微分算子,α和β是正参数;
在灰度图H的边缘,边缘检测函数ψ趋向于0,而在图像的平滑区域它趋向于1,考虑JEI-UDCP=0,根据EI-UDCP先验,得到去雾图像的景深d0=-logt0,其中透射率t0可表为:
Figure QLYQS_9
根据式(1)有:
Figure QLYQS_10
利用JEI-UDCP=0,易得透射率t0表达式。
3.根据权利要求1所述的基于边缘检测函数变分模型的水下图像去雾方法,其特征在于,所述步骤2中,
考虑如下的变分问题:
Figure QLYQS_11
其中
Figure QLYQS_12
α1,γ及λ=(λ01)为正参数,BV(Ω;R3)和
Figure QLYQS_13
分别代表向量值有界全变差空间和二阶有界广义全变差空间,离散形式的/>
Figure QLYQS_14
表为如下的最小值形式:
Figure QLYQS_15
最小值在Ω上的所有实向量场ω中取,并且
Figure QLYQS_16
代表对称导,研究变分模型的等价形式:
Figure QLYQS_17
4.根据权利要求1所述的基于边缘检测函数变分模型的水下图像去雾方法,其特征在于,所述步骤3中,
定义1:定义BV(Ω,R3)为包含满足下面条件的g的函数空间:
(1)g∈L1(Ω;R3);
(2)
Figure QLYQS_18
取有限值,BV(Ω;R3)是一个Banach空间,有范数:/>
Figure QLYQS_19
定义2:令
Figure QLYQS_20
为有界域且(λ01),其中λ01均为正数,对于/>
Figure QLYQS_21
带权值λ的二阶广义全变差(TGV of order two)定义如下:
Figure QLYQS_22
其中,
Figure QLYQS_23
为仅在Ω上,连续可微的对称矩阵域空间,Sn×n表示所有n×n的对称矩阵,空间/>
Figure QLYQS_24
有范数:/>
Figure QLYQS_25
称作为向量值函数的二阶有界广义全变差空间(BGV of order two);
引理1:假设gk∈BV(Ω;R3)(k=1,2,..),并且在
Figure QLYQS_26
上gk→g,那么
Figure QLYQS_27
引理2:假设dk∈BGV(Ω;R3)(k=1,2,..),并且在
Figure QLYQS_28
上dk→d,那么
Figure QLYQS_29
引理3:假设
Figure QLYQS_30
是一列定义在BV(Ω;R3)上,满足/>
Figure QLYQS_31
于是存在一个收敛子列/>
Figure QLYQS_32
和一个函数g∈BV(Ω;R3),使得/>
Figure QLYQS_33
引理4:假设
Figure QLYQS_34
是一列定义在BGV(Ω;R3)上,满足/>
Figure QLYQS_35
于是存在一个收敛子列/>
Figure QLYQS_36
和一个函数d∈BGV(Ω;R3),使得/>
Figure QLYQS_37
把变分问题的能量泛函写为E(g,d);
定理1:令f∈L2(Ω;R3)且d0∈L2(Ω;R3),于是对于固定的参数α1,γ>0,能量泛函有唯一的最小化方法。
5.根据权利要求1所述的基于边缘检测函数变分模型的水下图像去雾方法,其特征在于,所述步骤4中,记输入图像的规模为M×N,令X=RMN为有限维实向量空间,定义了内积<·,·>X
Figure QLYQS_38
其中梯度/>
Figure QLYQS_39
是向量空间Y=X×X上的向量;/>
Figure QLYQS_40
算子符合Neumann边界条件的离散形式定义为:
Figure QLYQS_41
其中,
Figure QLYQS_42
定义:
P1={p1∈R6MN:p1≤α1},P2={p2∈R6MN:p2≤λ1},Q={q∈R12MN:q≤λ0},
并且记闭凸集C上的指示函数IC为:
Figure QLYQS_43
对于一个真凸并且下半连续的函数Φ,定义近端算子
Figure QLYQS_44
为:
Figure QLYQS_45
于是变分问题等价形式可以重新写作下面的鞍点问题:
Figure QLYQS_46
其中p1,p2,q是对偶变量;
通过原始对偶算法,任给一个第k步的中间结果
Figure QLYQS_47
qk,gk,dkk,更新解的过程如下:
①更新变量
Figure QLYQS_48
qk+1,给定gk,dkk,更新/>
Figure QLYQS_49
qk+1为:
Figure QLYQS_50
Figure QLYQS_51
Figure QLYQS_52
其中
Figure QLYQS_53
②更新变量dk+1,gk+1k+1,记对称导的伴随算子为:
Figure QLYQS_54
其中
Figure QLYQS_55
给定/>
Figure QLYQS_56
qk+1,更新更新变量dk+1,gk+1k+1为:
Figure QLYQS_57
Figure QLYQS_58
Figure QLYQS_59
其中,
Figure QLYQS_60
进一步,得到:
Figure QLYQS_61
最终得到了水下去雾算法流程如下:
重复迭代:更新
Figure QLYQS_62
qk+1,gk+1,dk+1,/>
Figure QLYQS_63
如下:
Figure QLYQS_64
Figure QLYQS_65
Figure QLYQS_66
Figure QLYQS_67
Figure QLYQS_68
Figure QLYQS_69
Figure QLYQS_70
Figure QLYQS_71
Figure QLYQS_72
迭代终止条件:
Figure QLYQS_73
6.根据权利要求5所述的基于边缘检测函数变分模型的水下图像去雾方法,其特征在于,所述步骤4中,
定理2:记
Figure QLYQS_74
设定正参数σ1212满足
Figure QLYQS_75
和/>
Figure QLYQS_76
于是得到算法产生的序列/>
Figure QLYQS_77
收敛到/>
Figure QLYQS_78
也就是鞍点问题的解。
7.根据权利要求6所述的基于边缘检测函数变分模型的水下图像去雾方法,其特征在于,所述步骤5中,
评价指标峰值信噪比为:
Figure QLYQS_79
结构相似性为:
Figure QLYQS_80
其中,u0和u分别表示原图像和去雾图像,
Figure QLYQS_81
和μu是相应的平均值,/>
Figure QLYQS_82
和σu代表它们的标准离差,/>
Figure QLYQS_83
是u0和u的协方差,此外c1、c2是常量。
CN202310412291.4A 2023-04-18 2023-04-18 一种基于边缘检测函数变分模型的水下图像去雾方法 Pending CN116433525A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202310412291.4A CN116433525A (zh) 2023-04-18 2023-04-18 一种基于边缘检测函数变分模型的水下图像去雾方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202310412291.4A CN116433525A (zh) 2023-04-18 2023-04-18 一种基于边缘检测函数变分模型的水下图像去雾方法

Publications (1)

Publication Number Publication Date
CN116433525A true CN116433525A (zh) 2023-07-14

Family

ID=87088755

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202310412291.4A Pending CN116433525A (zh) 2023-04-18 2023-04-18 一种基于边缘检测函数变分模型的水下图像去雾方法

Country Status (1)

Country Link
CN (1) CN116433525A (zh)

Cited By (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116823677A (zh) * 2023-08-28 2023-09-29 创新奇智(南京)科技有限公司 一种图像增强的方法、装置、存储介质及电子设备

Cited By (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN116823677A (zh) * 2023-08-28 2023-09-29 创新奇智(南京)科技有限公司 一种图像增强的方法、装置、存储介质及电子设备
CN116823677B (zh) * 2023-08-28 2023-11-10 创新奇智(南京)科技有限公司 一种图像增强的方法、装置、存储介质及电子设备

Similar Documents

Publication Publication Date Title
CN108986050B (zh) 一种基于多分支卷积神经网络的图像和视频增强方法
CN112288658B (zh) 一种基于多残差联合学习的水下图像增强方法
CN112233038B (zh) 基于多尺度融合及边缘增强的真实图像去噪方法
CN109754377B (zh) 一种多曝光图像融合方法
CN109118446B (zh) 一种水下图像复原及去噪方法
CN111161360B (zh) 基于Retinex理论的端到端网络的图像去雾方法
CN111882489A (zh) 用于水下图像同时增强的超分辨率图形恢复方法
CN111861896A (zh) 一种面向uuv的水下图像色彩补偿与恢复方法
CN113284070A (zh) 基于传递注意力机制的非均匀雾图去雾算法
Zhang et al. Hierarchical attention aggregation with multi-resolution feature learning for GAN-based underwater image enhancement
CN113160286A (zh) 一种基于卷积神经网络的近红外和可见光图像融合方法
CN116433525A (zh) 一种基于边缘检测函数变分模型的水下图像去雾方法
CN113284061A (zh) 一种基于梯度网络的水下图像增强方法
CN114119383A (zh) 一种基于多特征融合的水下图像复原方法
CN113763300A (zh) 一种联合深度上下文与卷积条件随机场的多聚焦图像融合方法
CN117611467A (zh) 一种能同时平衡不同区域细节和亮度的低光图像增强方法
CN116823662A (zh) 一种融合原生特征的图像去噪去模糊方法
CN114820395B (zh) 一种基于多领域信息融合的水下图像增强方法
CN116630198A (zh) 一种结合自适应伽马校正的多尺度融合水下图像增强方法
Xie et al. Image defogging method combining light field depth estimation and dark channel
CN115908178A (zh) 一种基于暗通道先验的水下图像复原方法
CN116797468A (zh) 基于软边重建的自校准深度曲线估计的低光照图像增强方法
CN113012067B (zh) 基于Retinex理论和端到端深度网络的水下图像复原方法
Shibata et al. Reflection removal using RGB-D images
Li et al. Model-based single image deep dehazing

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination