CN114913083A - 一种基于上下文分解特征融合的水下图像增强方法 - Google Patents
一种基于上下文分解特征融合的水下图像增强方法 Download PDFInfo
- Publication number
- CN114913083A CN114913083A CN202210430466.XA CN202210430466A CN114913083A CN 114913083 A CN114913083 A CN 114913083A CN 202210430466 A CN202210430466 A CN 202210430466A CN 114913083 A CN114913083 A CN 114913083A
- Authority
- CN
- China
- Prior art keywords
- image enhancement
- underwater image
- feature fusion
- underwater
- context
- 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
Links
- 230000004927 fusion Effects 0.000 title claims abstract description 73
- 238000000354 decomposition reaction Methods 0.000 title claims abstract description 57
- 238000000034 method Methods 0.000 title claims abstract description 45
- 238000012549 training Methods 0.000 claims abstract description 30
- 238000012360 testing method Methods 0.000 claims abstract description 14
- 238000005070 sampling Methods 0.000 claims description 20
- 230000004913 activation Effects 0.000 claims description 10
- 230000008569 process Effects 0.000 claims description 10
- 238000011176 pooling Methods 0.000 claims description 8
- 230000007246 mechanism Effects 0.000 claims description 5
- 238000010276 construction Methods 0.000 claims description 3
- 230000000694 effects Effects 0.000 abstract description 8
- 238000012545 processing Methods 0.000 abstract description 7
- 230000002411 adverse Effects 0.000 abstract description 5
- 238000004458 analytical method Methods 0.000 abstract description 3
- 239000010410 layer Substances 0.000 description 48
- 230000006870 function Effects 0.000 description 19
- 238000005516 engineering process Methods 0.000 description 5
- 238000010586 diagram Methods 0.000 description 4
- 230000008878 coupling Effects 0.000 description 3
- 238000010168 coupling process Methods 0.000 description 3
- 238000005859 coupling reaction Methods 0.000 description 3
- 238000004891 communication Methods 0.000 description 2
- 238000001514 detection method Methods 0.000 description 2
- 238000005457 optimization Methods 0.000 description 2
- 230000006378 damage Effects 0.000 description 1
- 238000011161 development Methods 0.000 description 1
- 239000006185 dispersion Substances 0.000 description 1
- 239000002355 dual-layer Substances 0.000 description 1
- 230000002708 enhancing effect Effects 0.000 description 1
- 230000007613 environmental effect Effects 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 230000003287 optical effect Effects 0.000 description 1
- 230000008092 positive effect Effects 0.000 description 1
- 238000011084 recovery Methods 0.000 description 1
- 238000012827 research and development Methods 0.000 description 1
- 230000000630 rising effect Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/048—Activation functions
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
- G06T5/90—Dynamic range modification of images or parts thereof
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/10—Image acquisition modality
- G06T2207/10024—Color image
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Molecular Biology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Artificial Intelligence (AREA)
- Evolutionary Computation (AREA)
- General Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Image Processing (AREA)
Abstract
本发明提供了一种基于上下文分解特征融合的水下图像增强方法,涉及水下图像增强技术领域,包括:获取失真的水下图像,形成训练集和测试集;构建基于上下文分解特征融合的水下图像增强网络模型;其中,所述水下图像增强网络模型在特征融合时采用上下文分解特征融合的方式;将训练集输入到构建好的所述水下图像增强网络模型中进行训练;将测试集图像输入到训练好的所述水下图像增强网络模型中,输出图像增强之后的高质量水下图像。本发明通过上下文特征信息融合的方式,扩展了图像处理上特征融合的方式,也为模型的局部和全局分析建立了纽带,改善了水下图像增强的效果,避免了如细节丢失、颜色不饱和、伪影等不利影响的出现。
Description
技术领域
本发明涉及水下图像增强技术领域,特别是涉及一种基于上下文分解特征融合的水下图像增强方法。
背景技术
随着陆地资源紧张、人口增多、环境破坏等问题日益突出,海洋资源的研究和开发正在世界范围内兴起,然而海洋信息的获取和处理成为了海洋资源开发的第一步。因此,水下图像处理技术受到了越来越多的关注,水下图像增强技术在改善水下图像质量中发挥着重要的作用,为后续人们对水下信息的处理提供了帮助。
在现有的水下图像增强技术中,采用基于注意力的多尺度水下图像增强网络进行水下图像增强是一种非常有效的方法,其先通过下采样进行多尺度特征提取,并在反卷积阶段将相同大小的特征层连接,最后引入了注意力机制模块。该方法使网络关注不同尺度的特征,增强整体颜色和细节的恢复。
但是上述方法在进行特征信息融合时局限于连接下采样和上采样的中间层,忽略了空间尺度中的场景级上下文信息,导致许多不利的影响,比如细节丢失、颜色不饱和、伪影等。
发明内容
有鉴于此,本发明的目的在于提出一种基于上下文分解特征融合的水下图像增强方法,并实现高质量水下图像的生成以及应用,以解决现有水下图像增强技术中在特征信息融合时因忽略空间尺度中的场景级上下文信息而导致不利影响的技术问题。
为此,本发明提供了以下技术方案:
一方面,本发明提供了一种基于上下文分解特征融合的水下图像增强方法,所述方法包括:
获取失真的水下图像,形成训练集和测试集;
构建基于上下文分解特征融合的水下图像增强网络模型;其中,所述水下图像增强网络模型以失真的水下图像为输入,输出增强之后的高质量水下图像,特征融合时采用上下文分解特征融合的方式,所述上下文分解特征融合包括:将下采样的特征分别与相同尺度下对应的上采样特征进行融合,将得到的结果进行级联传到下一层;
将训练集输入到构建好的所述水下图像增强网络模型中进行训练;
将测试集图像输入到训练好的所述水下图像增强网络模型中,输出图像增强之后的高质量水下图像。
进一步地,所述水下图像增强网络模型的网络结构包括:
由两个Unet网络结构组成的双层估计机制,两个过程同时优化图像;UNet网络结构包含10个基本块,每个块由3个卷积层组成,其中每个下采样阶段包括两个Conv+BN+ReLU层和一个Max Pooling层,每个上采样阶段包括一个Up Sampling层和两个Conv+ReLU层。
进一步地,在特征融合时,对下层Unet网络结构的相同尺度下的特征分别与上层Unet网络结构的对应上采样特征执行乘法操作,然后将得出的结果进行级联操作传到下一层,从而实现上下文分解特征的融合。
进一步地,Unet网络结构中还包括:通道注意力模块和像素注意力模块。
进一步地,在通道注意力模块中,使用全局平均池化、两个卷积层和激活函数层将全局信息转化为通道权重信息,使不同通道特征具有不同的加权信息。
进一步地,在像素注意力模块中,使用两个卷积层和一个激活函数层将每个像素值加权。
进一步地,所述水下图像增强网络模型在训练时采用的损失函数包括:L1损失函数和对比度损失函数。
又一方面,本发明还提供了一种基于上下文分解特征融合的水下图像增强装置,所述装置包括:
图像获取单元,用于获取失真的水下图像,形成训练集和测试集;
模型构建单元,用于构建基于上下文分解特征融合的水下图像增强网络模型;其中,所述水下图像增强网络模型以失真的水下图像为输入,输出增强之后的高质量水下图像,特征融合时采用上下文分解特征融合的方式,所述上下文分解特征融合包括:将下采样的特征分别与相同尺度下对应的上采样特征进行融合,将得到的结果进行级联传到下一层;
训练单元,用于将训练集输入到构建好的水下图像增强网络模型中进行训练;
输出单元,用于将测试集图像输入到训练好的水下图像增强网络模型中,输出图像增强之后的高质量水下图像。
进一步地,所述水下图像增强网络模型的网络结构包括:
由两个Unet网络结构组成的双层估计机制,两个过程同时优化图像;UNet网络结构包含10个基本块,每个块由3个卷积层组成,其中每个下采样阶段包括两个Conv+BN+ReLU层和一个Max Pooling层,每个上采样阶段包括一个Up Sampling层和两个Conv+ReLU层。
又一方面,本发明还提供了一种计算机可读存储介质,该计算机可读存储介质内存储有计算机指令集,计算机指令集被处理器执行时实现上述基于上下文分解特征融合的水下图像增强方法。
本发明的优点和积极效果:本发明中,通过上下文特征信息融合的方式,扩展了图像处理上特征融合的方式,也为模型的局部和全局分析建立了纽带,改善了水下图像增强的效果,避免了如细节丢失、颜色不饱和、伪影等不利影响的出现。其次,本发明中的水下图像增强技术也将推动相关的二维图像处理与应用,比如水下目标检测、识别、重建等应用。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图做以简单地介绍,显而易见地,下面描述中的附图是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动性的前提下,还可以根据这些附图获得其他的附图。
图1是本发明实施例中基于上下文分解特征融合的水下图像增强的流程图;
图2是本发明实施例中基于上下文分解特征融合的水下图像增强的网络结构图;
图3是本发明实施例中上下文分解特征融合模块的结构示意图;
图4是本发明实施例中特征注意力模块的结构示意图。
具体实施方式
为了使本技术领域的人员更好地理解本发明方案,下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分的实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都应当属于本发明保护的范围。
需要说明的是,本发明的说明书和权利要求书及上述附图中的术语“第一”、“第二”等是用于区别类似的对象,而不必用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换,以便这里描述的本发明的实施例能够以除了在这里图示或描述的那些以外的顺序实施。此外,术语“包括”和“具有”以及他们的任何变形,意图在于覆盖不排他的包含,例如,包含了一系列步骤或单元的过程、方法、系统、产品或设备不必限于清楚地列出的那些步骤或单元,而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它步骤或单元。
本发明针对水下图像出现的色偏和色散现象,提出一种基于上下文分解特征融合的水下图像增强方法,可应用于水下导航,水下目标检测、识别和海底资源探索中。
如图1所示,本发明实施例中的一种基于上下文分解特征融合的水下图像增强方法,包括以下具体步骤:
S1、获取失真的水下图像,形成训练集和测试集;
S2、构建基于上下文分解特征融合的水下图像增强网络模型;
其中,水下图像增强网络模型以失真的水下图像为输入,输出增强之后的高质量水下图像,特征融合时采用的是上下文分解特征融合的方式,具体包括:将下采样的特征分别与相同尺度下对应的上采样特征进行融合,将得到的结果进行级联传到下一层。通过这种上下文分解特征的融合,可以充分挖掘多尺度上的上下文依赖关系,将上下文信息进行聚合,从而得到更有价值的特征信息,以便准确估计出图像,可以实现图像具有生动的颜色、足够的细节和很少的噪声。
S3、将训练集输入到构建好的水下图像增强网络模型中进行训练;
通过有限次的迭代学习以判断损失是否达到最小值,并且在学习过程中也不断地反向传播优化参数;在网络损失方面,本发明实施例中采用L1损失函数和对比度损失函数,该损失函数可以在训练阶段约束生成的图像和高质量的水下图像越来越接近。
其中,X,Y分别为模糊的水下图像和清晰的水下图像,N(X)为输入图像经过网络后输出的图像。
其中,V为VGG网络的第ReLU 5_3层。
经过训练之后,将失真的水下图像输入到网络中,图像将实现在保持正常颜色的同时减少模糊程度,提高图像的对比度。
S4、将测试集图像输入到训练好的水下图像增强网络模型中,输出图像增强之后的高质量水下图像。
为了便于理解,下面对上述基于上下文分解特征融合的水下图像增强网络模型进行详细说明。
上述基于上下文分解特征融合的水下图像增强网络模型的网络结构图如图2所示,为了实现上下文分解特征的融合,引入了由两个UNet结构组成的双层估计机制,两个过程同时优化图像。Unet的编码器用以提取目标特征,减少参数量,Unet的解码器用以重建特征并逐步恢复图像结构和纹理细节信息。UNet的网络结构包含10个基本块,每个块由3个卷积层组成,其中每个下采样阶段都包括两个Conv+BN+ReLU层和一个Max Pooling层,每个上采样阶段都包括一个Up Sampling层和两个Conv+ReLU层。在特征融合时采用上下文分解特征融合的方式,以上层Unet网络解码器学习到的潜在空间特征为指导,将其与下层Unet网络提取的多尺度特征相融合,这有助于实现全局信息和局部信息的增强。具体地,如图3所示,上下文分解特征融合模块(即图2中的MSP),包括:对下层Unet网络结构的相同尺度下的特征分别与上层Unet网络结构的对应上采样特征执行乘法操作,然后将得出的结果进行级联操作传到下一层。
双层Unet网络可以表示为:
上下文分解特征融合可以表示为:
为了充分挖掘隐藏在水下图像背后的真实颜色信息,使水下图像增强网络对不同的通道特征、不同的像素特征进行加权,关注到更重要的区域,比如图像的模糊区域。本发明在Unet的最小尺度加入了特征增强模块,包括通道注意力模块和像素注意力模块,在水下图像增强网络中,这两个模块可以自适应地学习不同特征的不同权重。
如图4所示,特征注意力模块(即图2中的UFEB):包括通道注意力和像素注意力。在通道注意力模块中,使用全局平均池化、两个卷积层和激活函数层将全局信息转化为通道权重信息,使不同通道特征具有不同的加权信息;在像素注意力模块中,使用两个卷积层和一个激活函数层将每个像素值加权。
通道注意力模块具体操作为:
A1=G(I);G为全局平均池化操作,I为模块的输入;
A2=CR(A1);CR为卷积+激活函数层;
A3=C(A2);C为卷积层;
A4=sigmoid(A3);sigmoid为激活函数;
像素注意力模块具体操作为:
A5=CR(A4);CR为卷积+激活函数层;
A6=C(A5);C为卷积层;
A7=sigmoid(A6);sigmoid为激活函数。
基于上下文分解特征融合的水下图像增强网络模型的网络整体的算法如下所示:
上述本发明实施例中,通过上下文特征信息融合的方式,扩展了图像处理上特征融合的方式,也为模型的局部和全局分析建立了纽带,改善了水下图像增强的效果,避免了如细节丢失、颜色不饱和、伪影等不利影响的出现。
对应本发明中的一种基于上下文分解特征融合的水下图像增强方法,本发明还提供了一种基于上下文分解特征融合的水下图像增强装置,包括:
图像获取单元,用于获取失真的水下图像,形成训练集和测试集;
模型构建单元,用于构建基于上下文分解特征融合的水下图像增强网络模型;
其中,水下图像增强网络模型以失真的水下图像为输入,输出增强之后的高质量水下图像,特征融合时采用的是上下文分解特征融合的方式,具体包括:将下采样的特征分别与相同尺度下对应的上采样特征进行融合,将得到的结果进行级联传到下一层。通过这种上下文分解特征的融合,可以充分挖掘多尺度上的上下文依赖关系,将上下文信息进行聚合,从而得到更有价值的特征信息,以便准确估计出图像,可以实现图像具有生动的颜色、足够的细节和很少的噪声。
训练单元,用于将训练集输入到构建好的水下图像增强网络模型中进行训练;
通过有限次的迭代学习以判断损失是否达到最小值,并且在学习过程中也不断地反向传播优化参数;在网络损失方面,采用L1损失函数和对比度损失函数,该损失函数可以在训练阶段约束生成的图像和高质量的水下图像越来越接近。经过训练之后,将失真的水下图像输入到网络中,图像将实现在保持正常颜色的同时减少模糊程度,提高图像的对比度。
输出单元,用于将测试集图像输入到训练好的水下图像增强网络模型中,输出图像增强之后的高质量水下图像。
对于本发明实施例的基于上下文分解特征融合的水下图像增强装置而言,由于其与上面实施例中的基于上下文分解特征融合的水下图像增强方法相对应,所以描述的比较简单,相关相似之处请参见上面实施例中基于上下文分解特征融合的水下图像增强方法部分的说明即可,此处不再详述。
本发明实施例还公开了一种计算机可读存储介质,该计算机可读存储介质内存储有计算机指令集,计算机指令集被处理器执行时实现如上文任一实施例所提供的基于上下文分解特征融合的水下图像增强方法。
在本发明所提供的几个实施例中,应该理解到,所揭露的技术内容,可通过其它的方式实现。其中,以上所描述的装置实施例仅仅是示意性的,例如所述单元的划分,可以为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些接口,单元或模块的间接耦合或通信连接,可以是电性或其它的形式。
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
另外,在本发明各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。
所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本发明的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的全部或部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可为个人计算机、服务器或者网络设备等)执行本发明各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、只读存储器(ROM,Read-Only Memory)、随机存取存储器(RAM,Random Access Memory)、移动硬盘、磁碟或者光盘等各种可以存储程序代码的介质。
最后应说明的是:以上各实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述各实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分或者全部技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的范围。
Claims (10)
1.一种基于上下文分解特征融合的水下图像增强方法,其特征在于,所述方法包括:
获取失真的水下图像,形成训练集和测试集;
构建基于上下文分解特征融合的水下图像增强网络模型;其中,所述水下图像增强网络模型以失真的水下图像为输入,输出增强之后的高质量水下图像,特征融合时采用上下文分解特征融合的方式,所述上下文分解特征融合包括:将下采样的特征分别与相同尺度下对应的上采样特征进行融合,将得到的结果进行级联传到下一层;
将训练集输入到构建好的所述水下图像增强网络模型中进行训练;
将测试集图像输入到训练好的所述水下图像增强网络模型中,输出图像增强之后的高质量水下图像。
2.根据权利要求1所述的一种基于上下文分解特征融合的水下图像增强方法,其特征在于,所述水下图像增强网络模型的网络结构包括:
由两个Unet网络结构组成的双层估计机制,两个过程同时优化图像;UNet网络结构包含10个基本块,每个块由3个卷积层组成,其中每个下采样阶段包括两个Conv+BN+ReLU层和一个Max Pooling层,每个上采样阶段包括一个Up Sampling层和两个Conv+ReLU层。
3.根据权利要求2所述的一种基于上下文分解特征融合的水下图像增强方法,其特征在于,在特征融合时,对下层Unet网络结构的相同尺度下的特征分别与上层Unet网络结构的对应上采样特征执行乘法操作,然后将得出的结果进行级联操作传到下一层,从而实现上下文分解特征的融合。
4.根据权利要求2所述的一种基于上下文分解特征融合的水下图像增强方法,其特征在于,Unet网络结构中还包括:通道注意力模块和像素注意力模块。
5.根据权利要求4所述的一种基于上下文分解特征融合的水下图像增强方法,其特征在于,在通道注意力模块中,使用全局平均池化、两个卷积层和激活函数层将全局信息转化为通道权重信息,使不同通道特征具有不同的加权信息。
6.根据权利要求4所述的一种基于上下文分解特征融合的水下图像增强方法,其特征在于,在像素注意力模块中,使用两个卷积层和一个激活函数层将每个像素值加权。
7.根据权利要求1所述的一种基于上下文分解特征融合的水下图像增强方法,其特征在于,所述水下图像增强网络模型在训练时采用的损失函数包括:L1损失函数和对比度损失函数。
8.一种基于上下文分解特征融合的水下图像增强装置,其特征在于,所述装置包括:
图像获取单元,用于获取失真的水下图像,形成训练集和测试集;
模型构建单元,用于构建基于上下文分解特征融合的水下图像增强网络模型;其中,所述水下图像增强网络模型以失真的水下图像为输入,输出增强之后的高质量水下图像,特征融合时采用上下文分解特征融合的方式,所述上下文分解特征融合包括:将下采样的特征分别与相同尺度下对应的上采样特征进行融合,将得到的结果进行级联传到下一层;
训练单元,用于将训练集输入到构建好的水下图像增强网络模型中进行训练;
输出单元,用于将测试集图像输入到训练好的水下图像增强网络模型中,输出图像增强之后的高质量水下图像。
9.根据权利要求8所述的一种基于上下文分解特征融合的水下图像增强装置,其特征在于,所述水下图像增强网络模型的网络结构包括:
由两个Unet网络结构组成的双层估计机制,两个过程同时优化图像;UNet网络结构包含10个基本块,每个块由3个卷积层组成,其中每个下采样阶段包括两个Conv+BN+ReLU层和一个Max Pooling层,每个上采样阶段包括一个Up Sampling层和两个Conv+ReLU层。
10.一种计算机可读存储介质,该计算机可读存储介质内存储有计算机指令集,计算机指令集被处理器执行时实现如权利要求1~7任一项所述的基于上下文分解特征融合的水下图像增强方法。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210430466.XA CN114913083A (zh) | 2022-04-22 | 2022-04-22 | 一种基于上下文分解特征融合的水下图像增强方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210430466.XA CN114913083A (zh) | 2022-04-22 | 2022-04-22 | 一种基于上下文分解特征融合的水下图像增强方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN114913083A true CN114913083A (zh) | 2022-08-16 |
Family
ID=82764997
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210430466.XA Pending CN114913083A (zh) | 2022-04-22 | 2022-04-22 | 一种基于上下文分解特征融合的水下图像增强方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114913083A (zh) |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116844192A (zh) * | 2023-07-19 | 2023-10-03 | 滁州学院 | 一种低质量指纹图像的增强处理方法 |
CN117876242A (zh) * | 2024-03-11 | 2024-04-12 | 深圳大学 | 眼底图像增强方法、装置、设备及可读存储介质 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111738948A (zh) * | 2020-06-19 | 2020-10-02 | 大连理工大学 | 一种基于双U-net的水下图像增强方法 |
CN112070688A (zh) * | 2020-08-20 | 2020-12-11 | 西安理工大学 | 一种基于上下文引导生成对抗网络的单幅图像去雾方法 |
CN112435191A (zh) * | 2020-11-25 | 2021-03-02 | 西安交通大学 | 一种基于多个神经网络结构融合的低照度图像增强方法 |
-
2022
- 2022-04-22 CN CN202210430466.XA patent/CN114913083A/zh active Pending
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111738948A (zh) * | 2020-06-19 | 2020-10-02 | 大连理工大学 | 一种基于双U-net的水下图像增强方法 |
CN112070688A (zh) * | 2020-08-20 | 2020-12-11 | 西安理工大学 | 一种基于上下文引导生成对抗网络的单幅图像去雾方法 |
CN112435191A (zh) * | 2020-11-25 | 2021-03-02 | 西安交通大学 | 一种基于多个神经网络结构融合的低照度图像增强方法 |
Non-Patent Citations (3)
Title |
---|
HAO-HSIANG YANG ET AL.: "LAFFNet: A Lightweight Adaptive Feature Fusion Network for Underwater Image Enhancement", 《2021 IEEE INTERNATIONAL CONFERENCE ON ROBOTICS AND AUTOMATION(ICRA)》, 18 October 2021 (2021-10-18) * |
LONG MA ET AL.: "Learning Deep Context-Sensitive Decomposition for Low-Light Image Enhancement", 《IEEE TRANSACTION ON NEURAL NETWORKS AND LEARNING SYSTEMS》, vol. 33, no. 10, 30 April 2021 (2021-04-30) * |
刘伟: "基于分层精简双线性注意力网络的水下细粒度目标识别方法研究", 《中国优秀硕士学位论文全文数据库 信息科技辑》, no. 03, 15 March 2022 (2022-03-15) * |
Cited By (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116844192A (zh) * | 2023-07-19 | 2023-10-03 | 滁州学院 | 一种低质量指纹图像的增强处理方法 |
CN116844192B (zh) * | 2023-07-19 | 2024-04-12 | 滁州学院 | 一种低质量指纹图像的增强处理方法 |
CN117876242A (zh) * | 2024-03-11 | 2024-04-12 | 深圳大学 | 眼底图像增强方法、装置、设备及可读存储介质 |
CN117876242B (zh) * | 2024-03-11 | 2024-05-28 | 深圳大学 | 眼底图像增强方法、装置、设备及可读存储介质 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN111243066B (zh) | 一种基于自监督学习与生成对抗机制的人脸表情迁移方法 | |
CN111754438B (zh) | 基于多分支门控融合的水下图像复原模型及其复原方法 | |
CN111832570A (zh) | 一种图像语义分割模型训练方法及系统 | |
CN114913083A (zh) | 一种基于上下文分解特征融合的水下图像增强方法 | |
Liu et al. | Effective image super resolution via hierarchical convolutional neural network | |
CN116309232B (zh) | 一种结合物理先验与深度学习的水下图像增强方法 | |
Wu et al. | FW-GAN: Underwater image enhancement using generative adversarial network with multi-scale fusion | |
CN111833261A (zh) | 一种基于注意力的生成对抗网络的图像超分辨率复原方法 | |
Zhou et al. | Domain adaptive adversarial learning based on physics model feedback for underwater image enhancement | |
Lin et al. | Autonomous underwater robot for underwater image enhancement via multi-scale deformable convolution network with attention mechanism | |
CN113762277A (zh) | 一种基于Cascade-GAN的多波段红外图像融合方法 | |
CN115100063A (zh) | 基于自监督的水下图像增强方法、装置及计算机存储介质 | |
Wang et al. | Underwater image super-resolution using multi-stage information distillation networks | |
Li et al. | Underwater Imaging Formation Model‐Embedded Multiscale Deep Neural Network for Underwater Image Enhancement | |
Wang et al. | A multi-level wavelet-based underwater image enhancement network with color compensation prior | |
Zhang et al. | A cross-scale framework for low-light image enhancement using spatial–spectral information | |
Zhang et al. | Enhanced visual perception for underwater images based on multistage generative adversarial network | |
Qiao et al. | Mutual channel prior guided dual-domain interaction network for single image raindrop removal | |
CN117036171A (zh) | 单幅图像的蓝图可分离残差平衡蒸馏超分辨率重建模型及方法 | |
Cheng et al. | FDCE-Net: underwater image enhancement with embedding frequency and dual color encoder | |
CN115511733A (zh) | 一种图像退化建模方法、神经网络训练方法和装置 | |
Fu et al. | Jointly adversarial networks for wavelength compensation and dehazing of underwater images | |
Wang et al. | Underwater image co-enhancement based on physical-guided transformer interaction | |
Tao et al. | Multi-scale Network with Attention Mechanism for Underwater Image Enhancement | |
CN112862698B (zh) | 一种基于结构感知和纹理敏感的物理启发学习的水下图像增强方法 |
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 |