CN110555814A - 图像去雾处理方法及装置、存储介质 - Google Patents

图像去雾处理方法及装置、存储介质 Download PDF

Info

Publication number
CN110555814A
CN110555814A CN201910817554.3A CN201910817554A CN110555814A CN 110555814 A CN110555814 A CN 110555814A CN 201910817554 A CN201910817554 A CN 201910817554A CN 110555814 A CN110555814 A CN 110555814A
Authority
CN
China
Prior art keywords
image
ambient atmospheric
atmospheric light
iteration
value
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.)
Withdrawn
Application number
CN201910817554.3A
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.)
Shenzhen Sensetime Technology Co Ltd
Original Assignee
Shenzhen Sensetime Technology Co Ltd
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 Shenzhen Sensetime Technology Co Ltd filed Critical Shenzhen Sensetime Technology Co Ltd
Priority to CN201910817554.3A priority Critical patent/CN110555814A/zh
Publication of CN110555814A publication Critical patent/CN110555814A/zh
Withdrawn legal-status Critical Current

Links

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
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20081Training; Learning
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20084Artificial neural networks [ANN]

Landscapes

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

Abstract

本申请公开了一种图像去雾处理方法及装置、存储介质,其中,所述方法包括:获取有雾图像;根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值;基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值;根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像。

Description

图像去雾处理方法及装置、存储介质
技术领域
本申请涉及计算机视觉技术领域,具体涉及一种图像去雾处理方法及装置、存储介质。
背景技术
图像去雾在许多领域有着重要的应用价值,例如,图像画质复原与增强、交通检测、环境监控等等。图像去雾算法涉及求解一个大气散射模型,只有有雾图像已知,而环境大气光,透射率和清晰图像均未知,这是一个高度病态不适定问题。目前的图像去雾算法的处理效果不尽如人意。
发明内容
本申请提供一种图像去雾处理的技术方案。
第一方面,本申请实施例提供了一种图像去雾处理所述方法,包括:获取有雾图像;根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值;基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值;根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像。
上述方案中,可选地,所述基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值,包括:按照变分模型的迭代格式,对所述有雾图像进行去雾处理,得到环境大气光、透射率及清晰图像的预测值;其中,所述变分模型的迭代格式表示为:
其中,
其中,An+1表示第n+1次迭代时环境大气光A的值,αA表示第n+1次迭代时环境大气光A的步长,tn+1表示第n+1次迭代时透射率t的值,αt表示第n+1次迭代时透射率t的步长,表示根据第n+1次迭代时环境大气光A的值An+1与透射率t的值tn+1得到的关于清晰图像J的中间解;αJ表示第n+1次迭代时清晰图像J的步长,Jn+1表示第n+1次迭代时清晰图像J的值。
上述方案中,可选地,所述方法还包括:基于大气散射模型确定变分模型,所述变分模型包括第一数据项、第二数据项、环境大气光约束项、透射率约束项和清晰图像约束项;结合梯度下降法和迭代法求解所述变分模型,得到所述变分模型的迭代格式。
上述方案中,可选地,所述大气散射模型表示为: 其中,I表示有雾图像,A表示环境大气光、t表示介质的透射率,J表示清晰图像。
上述方案中,可选地,所述变分模型表示为:
其中,表示约束条件;φ(A)表示环境大气光A的先验项,λ表示φ(A)的权重,ψ(t)表示透射率t的先验项,γ表示ψ(t)的权重,ω(J)表示清晰图像J的先验项,η表示ω(J)的权重,表示第一数据项和第二数据项之和;
其中, 表示第一数据项,表示第二数据项。
上述方案中,可选地,所述图像去雾处理方法通过图像去雾网络实现,所述方法还包括:利用所述图像去雾网络对训练样本中的有雾图像进行迭代去雾处理,得到与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值;根据与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值,以及训练样本中与所述有雾图像对应的清晰图像真值、环境大气光真值和透射率真值,确定环境大气光的损失、透射率的损失和清晰图像的损失;根据所述环境大气光的损失、所述透射率的损失和所述清晰图像的损失,结合损失函数确定总损失值;根据所述总损失值调整所述图像去雾网络的权重参数。
上述方案中,可选地,所述图像去雾网络包括所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络;所述根据所述总损失值调整所述图像去雾网络的权重参数,包括:基于所述总损失值调整所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络的权重参数。
上述方案中,可选地,所述根据所述总损失值调整所述图像去雾网络的权重参数,包括:确定每次迭代时的总损失值;每次迭代结束时,基于每次迭代时的总损失值,调整所述图像去雾网络的权重参数。
上述方案中,可选地,所述损失函数表示为:
其中,
Θn表示迭代n时的网络参数,表示训练样本为m的训练集,其中分别表示环境大气光、透射率和清晰图像的真值,分别表示环境大气光、透射率和清晰图像的预测值。
第二方面,本申请实施例提供一种图像去雾处理装置,所述装置包括:获取模块,用于获取有雾图像;第一处理模块,用于获取有雾图像;根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值;基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值;根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像。
上述方案中,可选地,所述第一处理模块,用于:按照变分模型的迭代格式,对所述有雾图像进行去雾处理,得到环境大气光、透射率及清晰图像的预测值;其中,所述变分模型的迭代格式表示为:
其中,
其中,An+1表示第n+1次迭代时环境大气光A的值,αA表示第n+1次迭代时环境大气光A的步长,tn+1表示第n+1次迭代时透射率t的值,αt表示第n+1次迭代时透射率t的步长,表示根据第n+1次迭代时环境大气光A的值An+1与透射率t的值tn+1得到的关于清晰图像J的中间解;αJ表示第n+1次迭代时清晰图像J的步长,Jn+1表示第n+1次迭代时清晰图像J的值。
上述方案中,可选地,所述装置还包括:第二处理模块,用于:基于大气散射模型确定变分模型,所述变分模型包括第一数据项、第二数据项、环境大气光约束项、透射率约束项和清晰图像约束项;结合梯度下降法和迭代法求解所述变分模型,得到所述变分模型的迭代格式。
上述方案中,可选地,所述大气散射模型表示为: 其中,I表示有雾图像,A表示环境大气光、t表示介质的透射率,J表示清晰图像。
上述方案中,可选地,所述变分模型表示为:
其中,表示约束条件;φ(A)表示环境大气光A的先验项,λ表示φ(A)的权重,ψ(t)表示透射率t的先验项,γ表示ψ(t)的权重,ω(J)表示清晰图像J的先验项,η表示ω(J)的权重,表示第一数据项和第二数据项之和;
其中, 表示第一数据项,表示第二数据项。
上述方案中,可选地,所述图像去雾处理方法通过图像去雾网络实现,所述装置还包括:第三处理模块,用于:利用所述图像去雾网络对训练样本中的有雾图像进行迭代去雾处理,得到与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值;根据与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值,以及训练样本中与所述有雾图像对应的清晰图像真值、环境大气光真值和透射率真值,确定环境大气光的损失、透射率的损失和清晰图像的损失;根据所述环境大气光的损失、所述透射率的损失和所述清晰图像的损失,结合损失函数确定总损失值;根据所述总损失值调整所述图像去雾网络的权重参数。
上述方案中,可选地,所述图像去雾网络包括所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络;所述第三处理模块,用于:基于所述总损失值调整所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络的权重参数。
上述方案中,可选地,所述第三处理模块,用于:确定每次迭代时的总损失值;每次迭代结束时,基于每次迭代时的总损失值,调整所述图像去雾网络的权重参数。
上述方案中,可选地,所述损失函数表示为:
其中,
Θn表示迭代n时的网络参数,表示训练样本为m的训练集,其中分别表示环境大气光、透射率和清晰图像的真值,分别表示环境大气光、透射率和清晰图像的预测值。
第三方面,本申请实施例提供了一种图像去雾处理装置,所述装置包括:存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,所述处理器执行所述程序时实现本申请实施例所述的图像去雾处理方法的步骤。
第四方面,本申请实施例提供了一种存储介质,所述存储介质存储有计算机程序,所述计算机程序被处理器执行时,使得所述处理器执行本申请实施例所述的图像去雾处理方法的步骤。
本申请提供的技术方案,获取有雾图像;根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值;基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值;根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像;不同于端到端的信息前馈传播的深度学习方法,结合信息前馈传播和迭代回传的过程,能有效地提高去雾效果。
附图说明
图1为本申请实施例提供的一种图像去雾网络的训练方法的实现流程示意图;
图2为本申请实施例提供的图像去雾算法概述框架示意图;
图3为本申请实施例提供的一种图像去雾处理方法的实现流程示意图;
图4为本申请实施例提供的一种图像去雾网络的训练装置的组成结构示意图;
图5为本申请实施例提供的一种图像去雾处理装置的组成结构示意图。
具体实施方式
这里将详细地对示例性实施例进行说明,其示例表示在附图中。下面的描述涉及附图时,除非另有表示,不同附图中的相同数字表示相同或相似的要素。以下示例性实施例中所描述的实施方式并不代表与本申请实施例相一致的所有实施方式。相反,它们仅是与如所附权利要求书中所详述的、本申请实施例的一些方面相一致的装置和方法的例子。
在本公开实施例使用的术语是仅仅出于描述特定实施例的目的,而非旨在限制本公开实施例。在本公开实施例和所附权利要求书中所使用的单数形式的“一种”、“一个”和“该”也旨在包括多数形式,除非上下文清楚地表示其他含义。还应当理解,本文中使用的术语“和/或”是指并包含一个或多个相关联的列出项目的任何或所有可能组合。
应当理解,尽管在本公开实施例可能采用术语第一、第二、第三等来描述各种信息,但这些信息不应限于这些术语。这些术语仅用来将同一类型的信息彼此区分开。例如,在不脱离本公开实施例范围的情况下,第一信息也可以被称为第二信息,类似地,第二信息也可以被称为第一信息。取决于语境,如在此所使用的词语“如果”及“若”可以被解释成为“在……时”或“当……时”或“响应于确定”。
下面结合附图和具体实施例对本申请的图像去雾处理的方案中用到的图像去雾网络的训练方法进行详细阐述。
本申请实施例提供一种图像去雾网络的训练方法,如图1所示,所述方法主要包括:
步骤S101、基于大气散射模型确定变分模型,所述变分模型包括第一数据项、第二数据项、环境大气光约束项、透射率约束项和清晰图像约束项。
其中,所述大气散射模型表示为:
式中,I表示有雾图像,A表示环境大气光、t表示介质的透射率,J表示清晰图像。
根据大气散射模型(1),图像去雾问题可以通过求解以下最小化问题:
其中,s.t.是such that的缩写,表示约束条件;其中,φ(·)和ψ(·)分别是关于环境大气光A和透射率t的先验;λ和γ是正的权重参数;表示数据项,通常定义为:
其中在本文中取为L2范数。
如果和φ(·),ψ(·)是可微的,则可用梯度下降法求解问题最小化问题(2)。在每次迭代中,我们需要求解
式中αA和αt分别表示A和t的步长;表示梯度算子;n=1,2,3…K-1表示迭代次数,K是最大迭代次数。
采用一种一步端到端的方式直接从I获得A和t,然后通过(1)计算J。由于这种双CNN结构能够产生合理的结果,我们将这种方法作为迭代法(4)的初始化方法,其公式如下:
其中,A1、t1和J1是(4)的初始值,N1在该初始化过程里是一个CNN。
如上所述,用基于梯度下降的优化方法(4)和初始化方法(5)迭代求解变分模型(2),可以解决图像去雾问题。
式(2)考虑了A和t上的先验项,可以直接得到J的闭形式解。然而,即使A和t可以较好地被估计,清晰图像J仍然缺乏适当的约束,由于图像去雾问题的高度不适定性,我们进一步引入J上的先验项来约束预测的清晰图像J,并将式(2)重新表述为变分模型,所述变分模型表示为:
其中,表示约束条件;φ(A)表示环境大气光A的先验项,λ表示φ(A)的权重,ψ(t)表示透射率t的先验项,γ表示ψ(t)的权重,ω(J)表示清晰图像J的先验项,η表示ω(J)的权重,表示数据项。
步骤S102、结合梯度下降法和迭代法求解所述变分模型,得到所述变分模型的迭代格式。
示例性地,所述迭代法为高斯-塞德尔(Gauss-Seidel)迭代法,或雅克比(Jacobi)迭代法,等等。
基于式(1),如果J的真值是已知的,则应尽可能相似于J的真值是已知的。尽管J的真值是未知的,预测的Jn是一个很好的近似值,因此它可以在下一个迭代中被用来更好地预测An+1和tn+1。通过这种方式,我们提出了A和t上的数据项:
其中,为简单起见,上述n和n+1被省略。采用类似的L2范数。在(3)中定义的数据项变成:
基于重新表述的目标函数(6)和数据项(8),用梯度下降法迭代求解J,并将(4)重新表述为:
其中,将式(8)中的记为第一数据项,将式(8)中的记为第二数据项。
我们用梯度下降法迭代求解J,并将(4)进行重新表述,可得所述变分模型的迭代格式表示为:
其中,An+1表示第n+1次迭代时环境大气光A的值,αA表示第n+1次迭代时环境大气光A的步长,tn+1表示第n+1次迭代时透射率t的值,αt表示第n+1次迭代时透射率t的步长,表示根据第n+1次迭代时环境大气光A的值An+1与透射率t的值tn+1得到的关于清晰图像J的中间解;αJ表示第n+1次迭代时清晰图像J的步长,Jn+1表示第n+1次迭代时清晰图像J的值。
(9)中的可微分先验项可通过以下公式学习:
其中,分别是以An、tn和Jn为输入的CNN。正权重参数λ、γ和η可以很自然地被一起学习。
步骤S103、将各种约束项的子网络嵌入到所述变分模型的迭代格式中,得到包含各种所述子网络的所述图像去雾网络。
步骤S104、利用训练样本训练所述图像去雾网络。
在一些可选实现方式中,所述利用训练样本训练所述图像去雾网络,包括:利用所述图像去雾网络对所述训练样本中的有雾图像进行去雾处理,得到与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值;根据与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值,以及所述训练样本中与所述有雾图像对应的清晰图像真值、环境大气光真值和透射率真值,确定环境大气光的损失、透射率的损失和清晰图像的损失;根据所述环境大气光的损失、所述透射率的损失和所述清晰图像的损失,结合所述损失函数确定总损失值;基于所述总损失值调整所述环境大气光约束项的子网络、所述透射率约束项的子网络、以及所述清晰图像的约束项的子网络的权重参数。
上述方案中,可选地,所述损失函数表示为:
其中,Θn表示迭代n时的网络参数,表示训练样本为m的训练集,其中分别表示环境大气光、透射率和清晰图像的真值,分别表示环境大气光、透射率和清晰图像的预测值。其中,
在一些可选实现方式中,所述利用训练样本训练所述图像去雾网络,包括:确定环境大气光、透射率和清晰图像的初始迭代值;从所述初始迭代值开始,在损失函数的监督下,通过所述变分模型的迭代格式,迭代优化环境大气光、透射率及清晰图像的解,直至达到最大迭代次数;根据损失函数确定最后一次迭代的总损失值;根据所述总损失值调整各种所述子网络的网络参数。
在一些可选实现方式中,所述从所述初始迭代值开始,在损失函数的监督下,通过所述变分模型的迭代格式,迭代优化环境大气光、透射率及清晰图像的解,包括:确定每一次迭代时环境大气光、透射率和清晰图像各项对应的预测值和真值;基于每一次迭代时环境大气光、透射率和清晰图像各项对应的预测值和真值,确定每一次迭代时环境大气光、透射率和清晰图像各项对应的损失;结合损失函数,确定每一次迭代时的总损失值;每次迭代结束时,基于每一次迭代时的总损失值,调整各项对应的约束项的子网络的权重参数。
本申请所述技术方案可用于各种图像去雾处理任务中,对场景不做限定,比如场景包括图像处理软件,交通检测,环境监控等。
本申请所述技术方案,用一般性的变分模型来刻画图像去雾问题,引入有效的数据项和先验约束项来约束模型,能改善不适定性;和以往的手工设计的图像统计先验的方法或者端到端的用卷积神经网络直接预测清晰图像的方法不同,本申请基于梯度下降法和迭代法来求解变分模型,用卷积神经网络嵌入到迭代算法有效地学习环境大气光、透射率和清晰图像的先验;不同于端到端的信息前馈传播的深度学习方法,结合信息前馈传播和迭代回传的过程,能有效地提高去雾效果。
在一些实施方式中,确定有雾图像I、每一次迭代时环境大气光、透射率和清晰图像各项对应的步长αA、αt、αJ,最大迭代次数K;第一次迭代时,基于得到所述有雾图像I的A1和t1,根据损失函数(13)和(14),以及A,t的真值,计算出A,t的总损失值,基于该总损失值调整A对应的初始化网络和t对应的初始化网络第二次迭代时,根据公式(9)得到A、t、J的预测值,根据损失函数(13)和(14)以及A、t、J的真值,计算出A、t、J的总损失值;根据总损失值,调整A对应的初始化网络和t对应的初始化网络以及A对应的约束项网络t对应的约束项网络J对应的约束项网络的权重参数。以此类推,直至迭代到最大次数K。
本申请所述技术方案可用于各种图像去雾处理任务中,对场景不做限定,比如场景包括图像处理软件,交通检测,环境监控等。
本申请所述技术方案,将图像去雾问题表述为一个变分模型的最小化问题,该模型具有良好的数据项和先验项,以使模型正则化;基于经典梯度下降法来求解这个最小化问题,深度CNNs被嵌入迭代过程中,这种方法可以很好地估计大气光、透射率和清晰图像的图像先验;结合图像去雾的物理模型和深度学习方法的特性,能够生成清晰的图像以及准确的大气光和透射率,有效地提高去雾效果。
在一种应用场景下,本申请实施例提供的图像去雾算法,该方法的概述框架如图2所示,该框架由两个关键组成部分组成:(a)、从(5)的初始化值开始,在阶段性损失函数(13)和(14)的监督下迭代优化中间解An、Tn和Jn,直至收敛到良好状态;(b)、迭代算法的基本结构(9),下一步的解An+1、Tn+1和Jn+1的学习过程被基于物理的数据项((3)和(7))和学到的An、Tn和Jn的先验项((10)、(11)和(12))约束。
实际应用中,将(5)在第一次迭代时表示为初始化方法,而将第n次(n=2,3···k)迭代时的迭代算法(9)表示为迭代单元。从初始化值开始,提出的深度图像去雾算法在若干次迭代中收敛。该方法采用迭代优化策略和有效的基于神经网络的学习方式,通过迭代优化一个变分模型并用深度CNNs来学习雾相关的图像先验,数据项和学到的深度图像先验有助于更好地估计大气光、透射率和清晰图像,采用基于梯度下降法的迭代方法数个迭代收敛。在基准数据集上进行的大量定性和定量实验表明,所提出的去雾处理算法表现良好。
对应地,本公开实施例提出了一种图像去雾处理方法,所述方法包括:
步骤S201:获取有雾图像。
步骤S202:根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值。
步骤S203:基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值。
步骤S204:根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像。
这里,所述预设次数大于或等于2。
在一些实施方式中,所述基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值,包括:按照变分模型的迭代格式,对所述有雾图像进行去雾处理,得到环境大气光、透射率及清晰图像的预测值。
其中,变分模型的迭代格式如上文式(9),在此不再赘述。
其中,变分模型如上文式(6),在此不再赘述。
上述图像去雾处理方法可通过图像去雾网络实现,所述方法还包括:利用所述图像去雾网络对训练样本中的有雾图像进行迭代去雾处理,得到与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值;根据与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值,以及训练样本中与所述有雾图像对应的清晰图像真值、环境大气光真值和透射率真值,确定环境大气光的损失、透射率的损失和清晰图像的损失;根据所述环境大气光的损失、所述透射率的损失和所述清晰图像的损失,结合损失函数确定总损失值;根据所述总损失值调整所述图像去雾网络的权重参数。
在一些可选实施方式中,所述图像去雾网络包括所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络;所述根据所述总损失值调整所述图像去雾网络的权重参数,包括:基于所述总损失值调整所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络的权重参数。
在一些可选实施方式中,所述根据所述总损失值调整所述图像去雾网络的权重参数,包括:确定每次迭代时的总损失值;每次迭代结束时,基于每次迭代时的总损失值,调整所述图像去雾网络的权重参数。
本实施例所述图像去雾处理方法,不同于端到端的信息前馈传播的深度学习方法,结合信息前馈传播和迭代回传的过程,能有效地提高去雾效果。
在一些实施例中,图像去雾网络获取有雾图像;确定最大迭代次数K;确定每次迭代时环境大气光、透射率和清晰图像各自对应的梯度下降法的步长α;基于所述迭代次数K和在每次迭代过程中环境大气光、透射率和清晰图像各自对应的所述步长α,利用图像去雾网络对所述有雾图像进行去雾处理,得到经过K次处理后的去雾图像;其中,所述图像去雾网络采用上文所述的方法训练。
对应上述图像去雾网络的训练方法,本申请实施例提供了一种图像去雾网络的训练装置,如图3所示,所述装置包括:第一确定模块10,用于基于大气散射模型确定变分模型,所述变分模型包括第一数据项、第二数据项、环境大气光约束项、透射率约束项和清晰图像约束项;第二确定模块20,用于结合梯度下降法和迭代法求解所述变分模型,得到所述变分模型的迭代格式;第三确定模块30,用于将各种约束项的子网络嵌入到所述变分模型的迭代格式中,得到包含各种所述子网络的所述图像去雾网络;网络训练模块40,用于利用训练样本训练所述图像去雾网络。
作为一种实施方式,所述网络训练模块40,用于:利用所述图像去雾网络对所述训练样本中的有雾图像进行去雾处理,得到与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值;根据与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值,以及所述训练样本中与所述有雾图像对应的清晰图像真值、环境大气光真值和透射率真值,确定环境大气光的损失、透射率的损失和清晰图像的损失;根据所述环境大气光的损失、所述透射率的损失和所述清晰图像的损失,结合所述损失函数确定总损失值;基于所述总损失值调整所述环境大气光约束项的子网络、所述透射率约束项的子网络、以及所述清晰图像的约束项的子网络的权重参数。
作为一种实施方式,所述网络训练模块40,用于:确定环境大气光、透射率和清晰图像的初始迭代值;从所述初始迭代值开始,在损失函数的监督下,通过所述变分模型的迭代格式,迭代优化环境大气光、透射率及清晰图像的解,直至达到最大迭代次数;根据损失函数确定最后一次迭代的总损失值;根据所述总损失值调整各种所述子网络的网络参数。
作为一种实施方式,所述网络训练模块40,用于:确定每一次迭代时环境大气光、透射率和清晰图像各项对应的预测值和真值;基于每一次迭代时环境大气光、透射率和清晰图像各项对应的预测值和真值,确定每一次迭代时环境大气光、透射率和清晰图像各项对应的损失;结合损失函数,确定每一次迭代时的总损失值;每次迭代结束时,基于每一次迭代时的总损失值,调整各项对应的约束项的子网络的权重参数。
上述方案中,所述大气散射模型表示为:其中,I表示有雾图像,A表示环境大气光、t表示透射率,J表示清晰图像。
上述方案中,所述变分模型表示为:
其中,表示约束条件;φ(A)表示环境大气光A的先验项,λ表示φ(A)的权重,ψ(t)表示透射率t的先验项,γ表示ψ(t)的权重,ω(J)表示清晰图像J的先验项,η表示ω(J)的权重,表示第一数据项和第二数据项之和;
其中, 表示第一数据项,表示第二数据项。
上述方案中,可选地,所述变分模型的迭代格式表示为:
其中,
其中,An+1表示第n+1次迭代时环境大气光A的值,αA表示第n+1次迭代时环境大气光A的步长,tn+1表示第n+1次迭代时透射率t的值,αt表示第n+1次迭代时透射率t的步长,表示根据第n+1次迭代时环境大气光A的值An+1与透射率t的值tn+1得到的关于清晰图像J的中间解;αJ表示第n+1次迭代时清晰图像J的步长,Jn+1表示第n+1次迭代时清晰图像J的值。
上述方案中,所述损失函数表示为:
其中,
Θn表示迭代n时的网络参数,表示训练样本为m的训练集,其中分别表示环境大气光、透射率和清晰图像的。
本领域技术人员应当理解,图3中所示的图像去雾网络的训练装置中的各处理模块的实现功能可参照前述图像去雾网络的训练方法的相关描述而理解。本领域技术人员应当理解,图3所示的图像去雾网络的训练装置中各处理单元的功能可通过运行于处理器上的程序而实现,也可通过具体的逻辑电路而实现。
实际应用中,上述第一确定模块10、第二确定模块20、第三确定模块30、和网络训练模块40的具体结构均可对应于处理器。所述处理器具体的结构可以为中央处理器(CPU,Central Processing Unit)、微处理器(MCU,Micro Controller Unit)、数字信号处理器(DSP,Digital Signal Processing)或可编程逻辑器件(PLC,Programmable LogicController)等具有处理功能的电子元器件或电子元器件的集合。其中,所述处理器包括可执行代码,所述可执行代码存储在存储介质中,所述处理器可以通过总线等通信接口与所述存储介质中相连,在执行具体的各单元的对应功能时,从所述存储介质中读取并运行所述可执行代码。所述存储介质用于存储所述可执行代码的部分优选为非瞬间存储介质。
本申请实施例提供的图像去雾网络的训练装置,能得到更好的去雾效果。
本申请实施例还记载了一种图像去雾网络的训练装置,所述装置包括:存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,所述处理器执行所述程序时实现前述任意一个技术方案提供的图像去雾网络的训练方法。
作为一种实施方式,所述处理器执行所述程序时实现:基于大气散射模型确定变分模型,所述变分模型包括第一数据项、第二数据项、环境大气光约束项、透射率约束项和清晰图像约束项;结合梯度下降法和迭代法求解所述变分模型,得到所述变分模型的迭代格式;将各种约束项的子网络嵌入到所述变分模型的迭代格式中,得到包含各种所述子网络的所述图像去雾网络;利用训练样本训练所述图像去雾网络。
本申请实施例提供的图像去雾网络的训练装置,能得到更好的去雾效果。
本申请实施例提供了一种图像去雾处理装置,如图4所示,所述装置包括:获取模块50,用于获取有雾图像;第一处理模块60,用于根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值;基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值;根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像。
在一些实施方式中,所述第一处理模块60,用于:按照变分模型的迭代格式,对所述有雾图像进行去雾处理;其中,所述变分模型的迭代格式表示为:
其中,
其中,An+1表示第n+1次迭代时环境大气光A的值,αA表示第n+1次迭代时环境大气光A的步长,tn+1表示第n+1次迭代时透射率t的值,αt表示第n+1次迭代时透射率t的步长,表示根据第n+1次迭代时环境大气光A的值An+1与透射率t的值tn+1得到的关于清晰图像J的中间解;αJ表示第n+1次迭代时清晰图像J的步长,Jn+1表示第n+1次迭代时清晰图像J的值。
在一些实施方式中,所述装置还包括:第二处理模块70,用于:基于大气散射模型确定变分模型,所述变分模型包括第一数据项、第二数据项、环境大气光约束项、透射率约束项和清晰图像约束项;结合梯度下降法和迭代法求解所述变分模型,得到所述变分模型的迭代格式。
在一些实施方式中,所述大气散射模型表示为: 其中,I表示有雾图像,A表示环境大气光、t表示介质的透射率,J表示清晰图像。
在一些实施方式中,所述变分模型表示为:
其中,表示约束条件;φ(A)表示环境大气光A的先验项,λ表示φ(A)的权重,ψ(t)表示透射率t的先验项,γ表示ψ(t)的权重,ω(J)表示清晰图像J的先验项,η表示ω(J)的权重,表示第一数据项和第二数据项之和;
其中, 表示第一数据项,表示第二数据项。
在一些实施方式中,所述图像去雾处理方法通过图像去雾网络实现,所述装置还包括:第三处理模块80,用于:利用所述图像去雾网络对训练样本中的有雾图像进行迭代去雾处理,得到与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值;根据与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值,以及训练样本中与所述有雾图像对应的清晰图像真值、环境大气光真值和透射率真值,确定环境大气光的损失、透射率的损失和清晰图像的损失;根据所述环境大气光的损失、所述透射率的损失和所述清晰图像的损失,结合损失函数确定总损失值;根据所述总损失值调整所述图像去雾网络的权重参数。
在一些实施方式中,所述图像去雾网络包括所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络;所述第三处理模块80,用于:基于所述总损失值调整所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络的权重参数。
在一些实施方式中,所述第三处理模块80,用于:确定每次迭代时的总损失值;每次迭代结束时,基于每次迭代时的总损失值,调整所述图像去雾网络的权重参数。
在一些实施方式中,所述损失函数表示为:
其中,
Θn表示迭代n时的网络参数,表示训练样本为m的训练集,其中分别表示环境大气光、透射率和清晰图像的真值,分别表示环境大气光、透射率和清晰图像的预测值。
实际应用中,上述获取模块50、第一处理模块60、第二处理模块70和第三处理模块80的具体结构均可对应于处理器。所述处理器具体的结构可以为CPU、MCU、DSP或PLC等具有处理功能的电子元器件或电子元器件的集合。其中,所述处理器包括可执行代码,所述可执行代码存储在存储介质中,所述处理器可以通过总线等通信接口与所述存储介质中相连,在执行具体的各单元的对应功能时,从所述存储介质中读取并运行所述可执行代码。所述存储介质用于存储所述可执行代码的部分优选为非瞬间存储介质。
本申请实施例提供的图像去雾处理装置,去雾效果更强,适应领域更广泛。
本申请实施例还记载了一种图像去雾处理装置,所述装置包括:存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,所述处理器执行所述程序时实现前述任意一个技术方案提供的图像去雾处理方法。
作为一种实施方式,所述处理器执行所述程序时实现:获取有雾图像;根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值;基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值;根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像。
本申请实施例还记载了一种计算机存储介质,所述计算机存储介质中存储有计算机可执行指令,所述计算机可执行指令用于执行前述各个实施例所述的图像去雾网络的训练方法。也就是说,所述计算机可执行指令被处理器执行之后,能够实现前述任意一个技术方案提供的图像去雾网络的训练方法。
本申请实施例还记载了一种计算机存储介质,所述计算机存储介质中存储有计算机可执行指令,所述计算机可执行指令用于执行前述各个实施例所述的图像去雾处理方法。也就是说,所述计算机可执行指令被处理器执行之后,能够实现前述任意一个技术方案提供的图像去雾处理方法。
本领域技术人员应当理解,本实施例的计算机存储介质中各程序的功能,可参照前述各实施例所述的图像去雾网络的训练方法的相关描述而理解。
在本申请所提供的几个实施例中,应该理解到,所揭露的设备和方法,可以通过其它的方式实现。以上所描述的设备实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,如:多个单元或组件可以结合,或可以集成到另一个系统,或一些特征可以忽略,或不执行。另外,所显示或讨论的各组成部分相互之间的耦合、或直接耦合、或通信连接可以是通过一些接口,设备或单元的间接耦合或通信连接,可以是电性的、机械的或其它形式的。
上述作为分离部件说明的单元可以是、或也可以不是物理上分开的,作为单元显示的部件可以是、或也可以不是物理单元;既可以位于一个地方,也可以分布到多个网络单元上;可以根据实际的需要选择其中的部分或全部单元来实现本实施例方案的目的。
另外,在本申请各实施例中的各功能单元可以全部集成在一个处理单元中,也可以是各单元分别单独作为一个单元,也可以两个或两个以上单元集成在一个单元中;上述集成的单元既可以采用硬件的形式实现,也可以采用硬件加软件功能单元的形式实现。
本领域普通技术人员可以理解:实现上述方法实施例的全部或部分步骤可以通过程序指令相关的硬件来完成,前述的程序可以存储于计算机可读取存储介质中,该程序在执行时,执行包括上述方法实施例的步骤;而前述的存储介质包括:移动存储设备、只读存储器(ROM,Read-Only Memory)、随机存取存储器(RAM,Random Access Memory)、磁碟或者光盘等各种可以存储程序代码的介质。
以上所述,仅为本申请的具体实施方式,但本申请的保护范围并不局限于此,任何熟悉本技术领域的技术人员在本申请揭露的技术范围内,可轻易想到变化或替换,都应涵盖在本申请的保护范围之内。因此,本申请的保护范围应以所述权利要求的保护范围为准。

Claims (10)

1.一种图像去雾处理方法,其特征在于,所述方法包括:
获取有雾图像;
根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值;
基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值;
根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像。
2.根据权利要求1所述的图像去雾处理方法,其特征在于,所述基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值,包括:
按照变分模型的迭代格式,对所述有雾图像进行去雾处理,得到环境大气光、透射率及清晰图像的预测值;
其中,所述变分模型的迭代格式表示为:
其中,
其中,An+1表示第n+1次迭代时环境大气光A的值,αA表示第n+1次迭代时环境大气光A的步长,tn+1表示第n+1次迭代时透射率t的值,αt表示第n+1次迭代时透射率t的步长,表示根据第n+1次迭代时环境大气光A的值An+1与透射率t的值tn+1得到的关于清晰图像J的中间解;J表示第n+1次迭代时清晰图像J的步长,Jn+1表示第n+1次迭代时清晰图像J的值。
3.根据权利要求1所述的图像去雾处理方法,其特征在于,所述方法还包括:
基于大气散射模型确定变分模型,所述变分模型包括第一数据项、第二数据项、环境大气光约束项、透射率约束项和清晰图像约束项;
结合梯度下降法和迭代法求解所述变分模型,得到所述变分模型的迭代格式。
4.根据权利要求3所述的图像去雾处理方法,其特征在于,所述大气散射模型表示为:
其中,I表示有雾图像,A表示环境大气光、t表示介质的透射率,J表示清晰图像。
5.根据权利要求3所述的图像去雾处理方法,其特征在于,所述变分模型表示为:
其中,表示约束条件;φ(A)表示环境大气光A的先验项,λ表示φ(A)的权重,ψ(t)表示透射率t的先验项,γ表示ψ(t)的权重,ω(J)表示清晰图像J的先验项,η表示ω(J)的权重,表示第一数据项和第二数据项之和;
其中, 表示第一数据项,表示第二数据项。
6.根据权利要求1至5任一项所述的图像去雾处理方法,其特征在于,所述图像去雾处理方法通过图像去雾网络实现,所述方法还包括:
利用所述图像去雾网络对训练样本中的有雾图像进行迭代去雾处理,得到与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值;
根据与所述有雾图像对应的清晰图像预测值、环境大气光预测值和透射率预测值,以及训练样本中与所述有雾图像对应的清晰图像真值、环境大气光真值和透射率真值,确定环境大气光的损失、透射率的损失和清晰图像的损失;
根据所述环境大气光的损失、所述透射率的损失和所述清晰图像的损失,结合损失函数确定总损失值;
根据所述总损失值调整所述图像去雾网络的权重参数。
7.根据权利要求6所述的图像去雾处理方法,所述图像去雾网络包括所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络;
所述根据所述总损失值调整所述图像去雾网络的权重参数,包括:
基于所述总损失值调整所述环境大气光的约束项的子网络、所述透射率的约束项的子网络、以及所述清晰图像的约束项的子网络的权重参数。
8.一种图像去雾处理装置,其特征在于,所述装置包括:
获取模块,用于获取有雾图像;
第一处理模块,用于根据所述有雾图像预测得到初始的环境大气光、透射率及清晰图像的预测值;基于上一次所得的环境大气光、透射率及清晰图像的预测值进行迭代,得到迭代后的环境大气光、透射率及清晰图像的预测值;根据每次迭代后的环境大气光、透射率及清晰图像的预测值对所述有雾图像进行去雾处理,直至处理次数达到预设次数后,输出最终的去雾图像。
9.一种图像去雾处理装置,所述装置包括:存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,其特征在于,所述处理器执行所述程序时实现权利要求1至7任一项所述的图像去雾处理方法。
10.一种存储介质,所述存储介质存储有计算机程序,所述计算机程序被处理器执行时,能够使得所述处理器执行权利要求1至7任一项所述的图像去雾处理方法。
CN201910817554.3A 2019-08-30 2019-08-30 图像去雾处理方法及装置、存储介质 Withdrawn CN110555814A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN201910817554.3A CN110555814A (zh) 2019-08-30 2019-08-30 图像去雾处理方法及装置、存储介质

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN201910817554.3A CN110555814A (zh) 2019-08-30 2019-08-30 图像去雾处理方法及装置、存储介质

Publications (1)

Publication Number Publication Date
CN110555814A true CN110555814A (zh) 2019-12-10

Family

ID=68738530

Family Applications (1)

Application Number Title Priority Date Filing Date
CN201910817554.3A Withdrawn CN110555814A (zh) 2019-08-30 2019-08-30 图像去雾处理方法及装置、存储介质

Country Status (1)

Country Link
CN (1) CN110555814A (zh)

Citations (6)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN106056557A (zh) * 2016-06-12 2016-10-26 南京邮电大学 一种基于改进大气散射模型的单幅图像快速去雾方法
CN106530240A (zh) * 2016-09-29 2017-03-22 南京邮电大学 一种基于多尺度融合和全变分优化的图像去雾方法
CN108182671A (zh) * 2018-01-25 2018-06-19 南京信息职业技术学院 一种基于天空区域识别的单幅图像去雾方法
CN108229525A (zh) * 2017-05-31 2018-06-29 商汤集团有限公司 神经网络训练及图像处理方法、装置、电子设备和存储介质
CN108537760A (zh) * 2018-06-07 2018-09-14 南京信息职业技术学院 一种基于大气散射模型的红外图像增强方法
CN110060215A (zh) * 2019-04-16 2019-07-26 深圳市商汤科技有限公司 图像处理方法及装置、电子设备和存储介质

Patent Citations (6)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN106056557A (zh) * 2016-06-12 2016-10-26 南京邮电大学 一种基于改进大气散射模型的单幅图像快速去雾方法
CN106530240A (zh) * 2016-09-29 2017-03-22 南京邮电大学 一种基于多尺度融合和全变分优化的图像去雾方法
CN108229525A (zh) * 2017-05-31 2018-06-29 商汤集团有限公司 神经网络训练及图像处理方法、装置、电子设备和存储介质
CN108182671A (zh) * 2018-01-25 2018-06-19 南京信息职业技术学院 一种基于天空区域识别的单幅图像去雾方法
CN108537760A (zh) * 2018-06-07 2018-09-14 南京信息职业技术学院 一种基于大气散射模型的红外图像增强方法
CN110060215A (zh) * 2019-04-16 2019-07-26 深圳市商汤科技有限公司 图像处理方法及装置、电子设备和存储介质

Non-Patent Citations (1)

* Cited by examiner, † Cited by third party
Title
赵胜楠等: "暗原色先验与NL-CTV模型相结合的图像去雾方法", 《计算机工程与应用》 *

Similar Documents

Publication Publication Date Title
US9858529B2 (en) Systems and methods for multi-task Bayesian optimization
JP7439151B2 (ja) ニューラルアーキテクチャ検索
WO2019111118A1 (en) Robust gradient weight compression schemes for deep learning applications
CN110210513B (zh) 数据分类方法、装置及终端设备
US20210182687A1 (en) Apparatus and method with neural network implementation of domain adaptation
CN110148088B (zh) 图像处理方法、图像去雨方法、装置、终端及介质
CA3137297C (en) Adaptive convolutions in neural networks
JP6718500B2 (ja) 生産システムにおける出力効率の最適化
JP2022549031A (ja) 正常なデータへのデータサンプルの変換
CN109885811B (zh) 文章风格转换方法、装置、计算机设备及存储介质
CN111723901A (zh) 神经网络模型的训练方法及装置
WO2020152233A1 (en) Action selection using interaction history graphs
CN114072809A (zh) 经由神经架构搜索的小且快速的视频处理网络
CN115880187A (zh) 基于去噪扩散概率模型的单图像反光去除方法及相关设备
CN111241843B (zh) 基于复合神经网络的语义关系推断系统和方法
CN112966754B (zh) 样本筛选方法、样本筛选装置及终端设备
CN116861262B (zh) 一种感知模型训练方法、装置及电子设备和存储介质
US20210125066A1 (en) Quantized architecture search for machine learning models
US20230360181A1 (en) Machine Learning for High Quality Image Processing
CN110555814A (zh) 图像去雾处理方法及装置、存储介质
Ren et al. Hardware implementation of KLMS algorithm using FPGA
US11416775B2 (en) Training robust machine learning models
JP7339219B2 (ja) 情報処理装置、情報処理方法、及びプログラム
CN117940951A (zh) 确定图像信号处理参数的方法、装置和感知系统
Haim et al. Deep Learned Phase Mask for Single Image Depth Estimation and 3D scanning

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
WW01 Invention patent application withdrawn after publication
WW01 Invention patent application withdrawn after publication

Application publication date: 20191210