基于渐进式双网络模型的低曝光图像增强方法 |
来源:一起赢论文网 日期:2021-10-01 浏览数:1100 【 字体: 大 中 小 大 中 小 大 中 小 】 |
第44 卷 第2期2021 年2 月计 算机 学 报CHINESEJOURNALOFCOMPUTERSVol .44No.2Feb. 2021基于渐进式双网络模型的低曝光图像增强方法黄淑英1 5胡 威2 )杨 勇2 )李红霞3 )汪 斌3 )1 :)(天津工业大学计算机科学与技术学院 天津 300 38 7)2 )( 江西财经大学信息管理学院 南昌 3300 32)3 )( 江西财经大学软件与物联网工程学院 南昌 330 03 2)摘 要 传统的图像增强方法对低曝光图像进行增强时, 通常只考虑到了亮度的提升, 忽略了增强过程中带来的噪声放大问题.而当前基于深度学习的方法利用端到端的网络直接学习低曝光图像到正常图像的映射关系, 忽略了低曝光图像形成的物理原理, 也没有考虑解决噪声放大的问题. 针对上述问题, 本文通过对图像降质的本质原因进行分析, 提出一种基于渐进式双网络模型的低曝光图像增强方法, 该方法包含图像增强模块以及图像去噪模块两个部分. 对每个模块的构建也采用了渐进式的思想, 考虑了图像由暗到亮的亮度变化, 以及从粗到细的图像恢复过程, 使增强后的结果更接近真实图像. 为了更好地训练网络, 本文构建了一种双向约束损失函数, 从图像降质模型的正反两个方向使网络学习结果逼近真实数据, 达到动态平衡.为了验证本文方法的有效性, 本文与一些主流的方法从主观和客观两方面进行了实验对比, 实验结果证明了本文方法得到的结果更接近真实图像, 获得了更优的性能指标.关键词 低曝光图像增强; 渐进式; 双网络; 双向约束损失函数中图法分类号TP391DOI号10. 1 189 7/SP.J. 101 6. 2021 .00384ALow-ExposureImageEnhancementBasedonProgressiveDualNetworkModelHUANGShu Yi ng15HUWei2)YANGYong2)LIHongXia3 )WANGBi n3 )1:){SchoolofComputerScienceandTechnologyTiangongUniversi ty?Tianjin300387)2){ SchoolofInformat ionManagement?JiangxiUni versi tyofFinanceandEconomi cs^Nanchang330032)3 ){SchoolofSoft wareandInierneiofThi ngsEngineering?JiangxiUni versi tyofFi nanceandEconomi cs?Nanchang330032)AbstractWhenthetraditi onali mageenhancementmethodsenhancedthelowexposurei mage,theyusual l yj ustconsi deredtheenhancementofbrightnessandignoredtheprobl emofnoiseamplification.Besi des,thecurrentdeepl earni ngmethodsusedtheendtoendnetworktodirectl yl earnthemappi ngrel ati onshipbetweenthel owexposurei mageandthenormali mage,ignori ngthephysicalpri ncipleoftheformati onofthel owexposureimage,anddi dnotconsi dersol vi ngtheprobl emofnoiseampli ficati on.Inordertosol vetheseprobl ems,thispaperpresentsal owexposurei mageenhancementmethodbasedonprogressivedualnetworkmodelbyanal yzi ngtheessentialcausesofimagedegradati on.Theproposedmethodincl udestwoparts :i mageenhancementmodul eandi magedenoisi ngmodul e.Theconstructi onofeachmodul eal soadoptstheprogressi vei deabyconsi deri ngthei magebrightnesschangefromdarktolightandthei magerestorati onfromcoarsetofine,sothattheenhancedresulti sclosertotherealimage.Furthermore,totrai nthenetworkbetter,abidirecti onal constrai ntl ossfuncti onisdesigned,whichmakesthel earni ngresul tofnetworkapproachtherealdatafrompositi veandnegati vedirecti onsofthe收稿日期:201 9 11 27; 在线发布日期:2020 05 18.本课题得到国家自然科学基金(61862030, 61662026) 、 江西省自然科学基金( 20182BCB2200 6, 20181BAB202010 , 20192ACB20002 , 2 0192ACBL2 1008) 资助. 黄淑英, 博士, 副教授, 主要研究方向为图像处理与机器学习. Email: shuyinghuang2010@126.com. 胡 威, 硕士研究生, 主要研究方向为图像去雾与图像复原. 杨 勇( 通信作者) , 博士, 教授, 中国计算机学会(CCF) 髙级会员, 主要研究领域为图像处理与机器学习. Email: greatyangy@12 6.com. 李红霞, 硕士研究生, 主要研究方向为深度学习与图像复原. 汪 斌, 硕士研究生, 主要研究方向为深度学习与图像复原.2 期黄淑英等: 基于渐进式双网络模型的低曝光图像增强方法385imagedegradationmodel,andfi nal lyachievesdynamicbalance.Experi mentalresul tsshowthattheproposedmethodismoreeffecti vethansomestate-of-the-artenhancementmethodsfrombothsubjecti veandobjecti veeval uations.Keywordslow-exposureimageenhancement;progressive;dualnetwork;bidirectionalconstrai ntlossfunctioni 引 言在夜间筹顏明不良的情况下获取到的?傳因为曝光不足. 往往导致圈像存在可见慶低、 细节. 不可见、噪声干扰大x 光照本均匀等问題?. 低鹿度的圈"像不仅在视觉上给观察者带来较差的感受, 而且会严重齋响基于图像倩息的计算机视觉系统的正常工作. 随着各种工业相机设备技术的发, 提■? 其奇辨率、曝光时间等性能都有了较大的改参? 但是刺用硬件提: 高分書率的方渎成本翁播化_力差,礙难做爾1:普及. 属此, 许多研究者开始运用图像增强的技术, 对图像进'行;后翁处理, 提高賀像的赓量《面1 为图像增强示例, 其中第一行为低曝光的图像* 第二行为采用.本文方法迸行亮度增强后的图像. 从结果可以着出,围像增强算法使低曝光图像在翕食、 细节等方面都得到了很大的提升, M此, 低曝光圈像增强技术布目标检测、智能骞驶、文通监控、遥感图像等领域具有十聋重荽的价ai 献曝光猶像(集一ter和本t方蠢磁强后的闺《(潷二管;) 示曹闺在过去的几十年间. 西像增强技术发展迅速4艮多方法栩继被提出用以解决低曝光倒像增强的问题. 巨前\针对图'像增强的方法主要分为三种T 分别是基于直方画均衡化的增强方法 、基于Reti nex理论的增强方法£69]以及近年来, 兴起的基于深度学习的增强方法3° 1 3 ]_基于直方两均衡化的增强方法主栗思想是將图像直方图统计的像素动态范围进行拉伸, 通过增加图像的对比度来达到提升视觉效果时目的. 这类增强算法主要包含基于爵部和基于全局的直方圈均衡化. 该类方法在处理前景和背景都过暗的匣像时能取得较好的效果, 同时由于计算量小, 该类方法在处理时间上也具有优势, 恒其往往因为对某些区域增强过度而会出现细节丢失和偏色等问题>针对直方1图均衡化增強方法存在的问蘧,一些改选算法又相继被提出来, Li 等人M提出了一种用于对低曝光图像进行对比度增強和去矂的新型虜像增强框架. 该框架利甩基于趙像素的局部自适, 座方抜进行去噪.在保待纹理细节的同时也可侧除噪声*同时采'用基于暗通道先验去雾方法的亮度#息自适应增强'参数, 尭服了图像过增强或欠增强的问M.Li n等入M提出了一种包括颜色通道拉伸、、 直方_平均和重映射的流水线方法, 能够取得较:好的结果, Sujee 等人&]利甩金宇搽M直方_四面的方法来増强图像对比度? 从而最大限度地提取圈像的債息^ 基于直方图均衡化的增强耸法虽然在时间上具有象大优势, 但由于在亮度增强的过程中考虑的是图像整体的统计特性, . 因此在处理复杂场;*图像时并不能取#狼好的效果.Rfttine: x理论是由Lairf14]?20世纪60 年:代雷次提出. 该連论的主要思想是人眼所感知到的物体雄顧色和鑫虞:由物体衰爾翁玟射特挫_: 人射特性(或光照特性) 共周作用决定. 其数学模型可表示为=El)在低曝光?像琅成过程中, s代表莸取到的低曝光图像.i?是正常曝光图像, L是照度圈像, 符号代表遂元素点乘?基于Keti nex理论的图像亮度增强算法的核心思想是首先估计照度: 西像L, 然后从低曝光_像S中去餘L奢量, 得到正常曝光的图像尺,其数学表达式为i?=?L1(23雜Land: 提出关于人纖视觉系统的Re¥in物理论 塞: TReti nex理论的_像增强方法被广泛研麗*基于尊 ■度Retinex理 , Jobson.A[ 1 ?:提:386 计導机攀报: _1苹进了一种多, R度的Retmex颜色恢复算法, 在迸行图像增强的同时保持颜色的一致性^Fu等人?结合Retmex理论提出了一种暴〒*融合的低曝光图像增:强方法>既保护了细节又提升了对比度, .受人类视觉系统的启 Ymg等人》7]设计了一种用于图像增強的多曝光融合框架,来握供准确的对比度和翕度増强?Guo等人[■提if)丁结构感知先验细化照度图, 并分孤在红、绿、1 蓝三个通道上计算以增强图像.尽營这些方法在某些情况下能取傳较好的结華, 但它们在对Reti nex模型中的反射孝和光照分解时仍然存在鳥暇性, 菌为设计一种适用于各种场景fit图像分解约束是图像增强中的主赛难点? 此外>, 由于求解反射率图是一个病态问題, 粗略的估计値容晷?导致图像增强的结果出现过曝光或欠曝光规象-随着人工智能的发廣及神经网祭赛爾像处理领域的优异表现. 基于深度学习的方法在图像增强颔域得到了迅速发展?Lore 等人1>]提出了一种基于.深度自编码的方法对低_光阁輟进行增强和去噪, 既提高了图像的亮度, 又避免了图像过曝光的情况.Chen等人 创建了一个根据相机曝光对间而产电的低曝光图像数据集, 并设计了一神全畚积的端到端两络来卖现图像的増强?Wd等人[竭将人类视觉理论和卷积神经.网絡结合, 把低曝光图像分解为反射率?和光照囹,-并利用增强网络实现菌像S度的增強?Jiang等人[ 2 1 ]曾次裘出利用非配对的低/正常亮度的图像来训练图像堯度增强网络, 这种训练策略消除了对成对训练数据的依赖. Wang等人[ 2 3]没有直, 接学习低亮度图像和m常亮度图像之间的映射, 而?晕估计菌像到光照菌之间的映射来增强曝光不足的图像, 该方法增強了网络学习对复杂图像的调整能力- 虽然上述方法都館获得较好的效果 是这些基于深度學习的方法都是端到端的直接:孿习低曝光 像到芷繪图像的映射关系, 忽略了低曝光图像形成的物理原理.同时没有显式地包含*噪过程或仅仅依赖予传统的去噪方法, 因而在图像增强后眘在细节丢失,和噪声放大等问题.基于J:述分析, 本文针对现有低曝光图像增强方■奪在凿 振性, 握:出:一稀藝手Retinex理论的渐迸式双网络低曝光图像增强■型(如图2i? 该网络以低曝光廚像作为输人. 和用不同尺度的卷积核进行特征提取, 最终学万到Reti nex獏塑中的照度菌, 接着將该照度图代人Retmex隹型中计算#到S度谱辑辰的顧像. 然辰针对增强过程中出现的噪声放大问题, 将增强后的图像再■经过一十图像去嵊网络s 得到最终的增强结果? 论文的创. 新工作如下j( 1) 提报一种渐进式时双网络低曝光图像增:强模型, 整个模型针对低曝光图像增強中的低S度和噪声放大问题, 利用惭进式的思想设计了图偉增强模块和图像去噪模块.(2:) 在图像增强樓望的两个模块内部, 分别采用渐迸式的思想构建其网络框架, 对靡像实现从粗到细的修复过程,以取得更好的增嬅结果-( 3> 考虑到图像降质理论的可逆性, 提出了一[ 低曝光醜『LF方IP照麵1 亮度初步重建醜IPIP照度图2亮度重_图像0: 除操作?: 拼接?: 减操作:32*384*384的卷积块:64*192*192的卷积块:128*96*96的卷积块图2 渐进式双网络低曝光图像增强模型黄淑英等: 基于渐进式双网络模型的低曝光图像增强方法 3872 期种双向约束损失函数用于网络学习, 从图像降质模型的正反两个方向进行损失计算, 使学习到的信息更完整.本文第2 节重点介绍本文所提出的方法; 第3节将实验方法与对比实验进行详细说明; 第4 节总结全文工作及对未来研究展望.2 渐进式图像增强方法2. 1 网络框架针对现有的方法在图像增强后存在细节丢失和噪声放大的问题, 本文提出一种渐进式的双网络低曝光图像增强模型, 该模型包括图像亮度增强模块和图像去噪模块两个子网络, 构建的整体框架如图2 所示. 我们将网络设计与图像增强的物理模型相结合, 首先采用渐进式的思想对低曝光图像实现亮度的增强, 然后针对增强后图像出现的噪声被加强的问题, 对亮度增强结果加人图像去噪过程. 值得一提的是, 为了取得更好的图像增强结果, 本文在图像增强和图像去噪两个模块内部都引人了渐进式的思想, 实现从粗到细的亮度增强和噪声的去除.图像去噪模块和增强模块不同的是, 图像噪声通常被认为一种加性噪声, 因此在图像去噪模块的最后采用了减法操作, 通过将输人图像直接减去学习到的噪声得到最终的输出结果. 为了更好地训练网络, 本文根据图像降质理论的可逆性, 构建了一种正反双向约束的损失函数来训练网络, 其具体内容将在2.2节进行详细介绍.2.1.1 图像亮度增强模块考虑到真实场景下亮度提升是一个由暗到亮的过程, 本文提出一种渐进式的图像增强模块, 对低曝光图像先后两次从粗到细地实现图像的亮度和色彩的提升, 来完成图像亮度增强的过程. 该网络模块的实现过程如下描述.如图2 的图像增强模块( 上端虚线框) 所示, 该模块的输人为低曝光图像, 模块中的两个子框架输出均为照度图L, 输人输出均为红、绿、蓝三个通道.该模块包含亮度初步重建和亮度加强两步, 第一步的框架中包含六个卷积层, 网络前两层利用卷积操作实现特征图的下采样, 该操作既可以保证下采样的效果, 同时也避免了下采样造成的信息丢失. 通过下采样减小特征图的尺寸, 可以扩大卷积核的感受野, 使原本只能学习到3*3 范围信息的卷积核在两次尺寸下降后可以学习到7*7 范围的信息. 此外,为了保证最后得到的照度图与输人图像的尺寸一样, 在网络中使用了2 次反卷积操作, 来实现尺寸增大的目的, 同时也方便梯度下降的计算. 第二步的亮度加强的网络框架与第一步类似, 这一步的目的是精细化增强后的图像. 该部分网络中只设置了四个卷积层, 通过以较少的网络训练参数, 来提高网络性能. 本文在3.3 节也给出了实验, 证明了使用四个卷积层比直接使用六个卷积层得到了更好的结果. 该模块为了保证网络的有效性及更好地利用原图信息, 首先将初步增强的结果与原图采用一次拼接操作, 然后再采用两次卷积和反卷积操作, 实现图像进一步增强. 在这些卷积层中, 每层都包含权重和偏差两类参数, 其计算方式如下式:Fix)=(〇*x ̄\ ̄b( 3)式中, F是卷积之后得到的特征图, 分别是权重和偏差, :r 是输人, 符号“”代表卷积计算.在整个框架中, 每个卷积层的后面都有一层激活函数ReLU, 其定义如下:R{x)=max( 0, F(x) )( 4)其中, _F(:r) 为卷积得到的结果,i?( :r) 是ReLU函数得到的结果. ReLU激活函数的目的是将有效信息保存的同时去除无效信息, 从而加快训练速度.2.1.2 图像去噪模块低曝光图像在成像过程中除了光照不足以外,还会受多种噪声影响. 在低亮度情况下由于图像的对比度比较低, 噪声难以被发现. 但经过对低曝光图像增强后, 噪声同样也会出现被增强的问题.因此,为了去除噪声, 本文在图像增强模块之后增加了一个噪声去除模块.如图2 的图像去噪模块(下端虚线框) 所示, 该模块的网络设计思想与增强模块类似, 采用渐进式的思想先后两次从粗到细来学习噪声图像, 实现对亮度重建图像的噪声去除. 在模块内部通过对多个尺寸的特征图卷积的方式来扩大感受视野, 学习更丰富的特征. 该网络模块的目的是学习图像中存在的噪声成分, 而通常图像中噪声被认为是一种加性噪声, 因此在两个子块后端采用了减法操作, 将增强后的图像与网络学习到的噪声成分相减, 得到去噪后的图像.由于该网络模块学习噪声成分并不需要考虑图像特征信息复用的问题, 因此在去噪网络模块中并没有采用类似于增强模块中的拼接操作.388 计 算机 学 报 2021年2. 2 损失函数网络模型的执行功能主要取决于损失函数的定义. 在图像复原的网络模型学习中, 通常使用均方误差MSE( MeanSquaredError) 或平均绝对误差MAE( MeanAbsol uteError) 之类的误差指标来定义损失函数. 但是, 由于低曝光图像亮度较低, 仅使用诸如MSE或MAE之类误差指标可能导致结构失真, 例如模糊效应或伪影. 所以为了提高视觉质量, 本文设计由结构损失和双向约束损失构建新损失函数来训练网络. 其中, 图像亮度增强模块的损失函数定义如下:Le ̄LgL/(5)式中, Ls为结构相似度损失函数, &为双向约束损失函数.图像去噪模块的损失函数定义为Ln=Lsp+Lsn(6)其中, 是正向结构损失函数, LS?是反向结构损失函数. 双向约束损失函数是从图像降质模型正反两个方向, 通过分别向真实数据逼近来定义的. 结构相似度损失函数是用来保证图像结构信息的完整性,具体定义将在下面给出.2.2.1 结构相似度损失函数在获取全局信息的同时, 网络还通过对多个尺寸的特征图卷积来学习结构细节信息, 因此我们用多尺度结构相似性MSSSJM( Mul ti ScaleStructualSmnl anty)[2 3]质量评价方法作为损失函数, 以保持图像结构, 避免模糊. 其获取方式为MS(ai , a2)=l (a1, az)?JJc(ai, a2) f , a2 Y- (7)其中, A、 ?为对应的图像输人, / (A,A) 为亮度信息,c 为对比度信息y为结构相似性,z 为像素坐标,?为像素总数, 《、/?、 7 为调整重要性的参数, ¥5的值越大, 则结构信息越完整, 结构相似度损失函数定义为Lms(aj, a2 )= l MS(ai ,a2)(8)因此, 基于上述式( 8) , 图像亮度增强模块的结构相似度损失函数Ls可以定义为Ls=LMS (yt , fc sy)(9)其中,_y 代表真实正常曝光图像, S为低曝光图像,/( S) 为网络输出结果.2.2.2 双向约束损失函数增强模块学习的目的是得到精确的照度图L=/( S), 再根据成像模型( 2) 计算得到增强后的图像i?=s*/(sr1. 传统的深度学习定义损失函数的方法, 主要是通过将网络输出与真实数据之间的差异最小化来训练网络, 这样可以让输出结果向真实数据快速接近. 因此, 本文也将考虑利用将图像增强结果与正常曝光图像之间差的绝对值最小化来训练增强网络模块, 其计算方法为l ^ ( S*/1(S) )!| (10)其中, 〃指的是样本的数量,z 指的是第z 个样本.根据式(1) , 我们也可以通过反推的方法来验证增强网络输出的照度图/(S) 的精确性, 即将/( S)代人到式⑴中, 用真实的图像与厂HS) 相乘, 得到的降质图像与网络输人的低曝光图像进行比较, 如果越接近网络的输人图像S, 则说明网络模型输出结果越精确. 基于该思想, 本文将反推过程引人到网络训练中, 我们将上述式( 10) 定义为正向推进损失,将反推过程定义为反向推进损失, 使网络不仅能正向逼近真实结果, 还能反向逼近输人, 达到动态平衡, 提尚网络准确率. 基于上述分析, 本论文提出一■种双向约束的损失函数, 其定义如下:Lf—LpLn(11)其中,L?=X\St (y^f(SY)t\ (12)表示反向推进损失.双向约束损失的思想不仅在图像增强模块中有效, 在去噪模块中也能起到同样的作用. 同样基于双向约束的思想, 本文定义正向与反向结构损失函数来训练图像的去噪模块, 其分别表示为Lsp=LMS(y, y?f( y?)')( 13)Ls?=LMS (y? , y+ f(y? ) )(14)其中, 是加噪的数据集,/(A) 为去噪网络学习得到的噪声.2.3网络训练2.3.1 训练方法由于增强模块和去噪模块是分开训练的, 所以在最小化损失函数的时候, 两个模块的损失函数也是不同的. 我们采用式( 5) 来训练图像增强模块, 用式(6) 训练图像去噪模块, 优化方式为Adam梯度下降方法. 整个网络的训练过程如算法1 所示.算法1.网络训练过程.输人: 单次训练样本数n输出: 照度图/(S), 去噪后图像步骤:fornum=1; numdterati onsdo低曝光图像集黄淑英等: 基于渐进式双网络模型的低曝光图像增强方法 3892 期输出增强图: 只―/(S) ;梯度下降法更新图像增强损失函数;endfornum=1;num么i t erat i onsdo加噪图像输出去噪图: 只'—/(A) ;梯度下降法更新图像去噪损失函数;end2.3.2网络设置整个网络共有20 个卷积层, 包含增强和去噪两个部分. 其中每个模块有三个采用步长为2 的卷积层以及3 个反卷积层, 在所有卷积层中, 除了3*3大小卷积核以外, 还采用了1*1 大小的卷积核, 来增强模型的非线性, 减少参数量, 提高计算速度. 其具体参数设置如表1 所示.网络部分图像亮度增强模块图像去噪模块卷积层输人123456输出输人1234输出表1 网络参数表输出尺寸卷积核数量 卷积核尺寸步长12212213 实验结果与分析我们将在合成图像数据集和真实场景图像数据集上验证本文提出方法的有效性, 并与其它六种图像增强方法进行比对, 这六种方法分别是:BIMEF[1 7]、Reti nexnet[20]、 UME[1 8]、MSRCR[1 5]、MF[ 1 6 ]、DONG[ 2 4]_此外, 对低曝光图像只经过第一个亮度增强模块后的结果与对亮度增强后图像进行去噪的结果也分别进行了展示, 来验证只对图像进行增强会产生噪声放大的问题, 进一步说明了本论文提出的渐进式的双网络模型的有效性. 相关的实验在3.3 节进行了说明.3. 1实验设置用于图像增强模块训练的数据库包括两个,一是低曝光图像对数据库( L0L)[2°], 其中包含了500 张低曝光图像和对应的正常曝光图像, 每张图像的大小为400*600.L0L的大部分图像都是在自然场景下通过调整曝光时间来获得的, 场景包括房屋、 郊区、 街道等, 本文方法将使用该数据库485 个图像对用于训练, 余下15 个图像对用于测试.另一个数据库为合成数据库, 该数据库从RAISE?数据库挑选1000 张原始图像, 用AdobeUghtroom提供的接口, 调整图像亮度通道Y的值,将这些图像合成低曝光图像, 得到1〇〇〇 对相应图像对. 最后, 将这些图像对调整为400*600 大小. 在对网络训练时, 为了方便计算, 所有输人都将调整为384*384大小.对于去噪网络, 由于现有带噪声的低曝光图像数据库缺乏, 所以我们结合低曝光图像噪声的特征,在增强模块训练集的正常曝光图像上进行了泊松噪声添加的处理, 以此作为去噪模块的训练数据, 图3为部分原始数据及对应加噪数据展示.在训练时, 上述LOL数据库及合成数据库所有图像都会作为训练数据, 出于方便计算, 输人与输出的像素值都会经过归一化操作变换到〇 至1 之间.为了避免在计算正常曝光图像时出现分母为零的现象, 我们将照度图的值设定在0.0039(1/255) 到1之间.网络学习率设置为〇.00001.所有实验都在Wi ndows10 系统、PyTorch 框架以及英伟达2080TiGPU上运行的.390 计 算机 学 报 2021年mmm,■■■■■■D_MD低曝光数据集正常曝光数据集(a) 部分低曝光数据集及对应的正常曝光数据集加噪数据集无噪数据集(b) 部分加噪数据集及对应的无噪清晰数据集, 第二、四列为局部放大示例图3 数据库实例3. 2 客观指标为了证明本文方法的性能优势, 我们采用峰值信噪比(PSiV^R) 、 结构相似度(SSJM)[26]&及色差值(CJ£:D£:2000)[27]H种指标来对各种方法进行客观评价.PSiVi? 是一种全参的图像质量评价指标, 计算每个像素点之间的差异, 其计算方法为p-10l<mk)個其中. 指的是图像中像素点的最大.僮,一般取25&MSE指的是.均方误#,MSE计算如下:IIWMSE=-^22)2(16)ijS中, 是图輟长宽,X、Y对应?像中的像素氇, W*j>綦闺像中修翥点的坐:_?PSIVi?_值越高^则图像质量越好.SSZM也是全参的图像质黧评价指标, 其值?分布在: 0 到1 之间准越大说明差异越小JJT面像赓璧越好? 其计算方法为SS—Z( %*jtCh. if*sicii )(17)其中,/xy分别表示两张廚片的盡犯、对比度、结构三If面的相似度,CJEDE2000 能够测试增强图像与真实数据之间的色差_ 霄先. 将RGB圈像转换到Lab 颜色會间, 参照CJEDE2000 计算色羞; 的方法得到两幅图像y和L对应的每个像素点之间的色齡AE 并按如下定义计箕两幅图像的色差:1Nc=j^XAE<yi上)<i?)其中, N是为像素数■, AE表示Lab 空间中w个圓镩之间的每个像素的色差 为网络输出图傳,L为真实数据? 色差值越低. 表示颜色越接近真实菌像.3. 3 实验结果在这一部分, 我们将展示本文方法对LOL数据集以及/合成数据库中部分菌像进行增强的着终结果? 主观效果好坏是人眼嚴直接的感. 受< 如?4 所示, 为本文方法与其它六种方:法在不同场景下搏到的结果■ 从图中可以奢出, BIMEF得到了较为猜晰的细节, 并且控制了噪声的扩张, 但暴其结某整体亮度偏低. DON:G在:亮度増强方面比较有效, 但其在围像增强后出现了嗓舞放大的问题, 如在第二列和第三列中的西像,白色框中的平滑废域s 可以着出噪声故大非常严重: ; 苘徉在第五列的图像中也出现了偏色问题: LIME和MSRCR方法在得到丰富细节信惠的同时, 产生T过曝的现象, 并且也出现了噪声放大的何题, 如白色柩:中区域可以观察勤明遽的过曝光情况*所有顧像的堯度都高于真实图像*同祥在:第五列中可以着到较严重的偏色问題. MF实现3*较.为6然的结杲, 但也伴M了轻徵的嗓声与偏色效果;R^tinex部&t 出现了较严重的色彩.失真和噪声敗大问题, 如平滑区域W以观察到'较产重的噪_拿,在白色框区域也可以看到明显的色彩失真. 与这些方法相比r我们的方法既实现了盡度增强,图像的色彩更自然, 也没有产生多佘的噪F, 其结果也最接近离实虜像.黄淑英等: 基于渐进式双网络模型的低曝光图像增强方法 39 1 2 期DONG本文真实图像图4 图像增强结果除了主观效果以外, 本文也利用了3.2 节中介绍的客观指标来对各方法进轩了比较. B5绐出了本文方法着其它六种方法. 对15 幅图像进行増强得到餘客.观搢标和eiEDF20: 00的、警均偉*: Qurs隹狄数据是本文方法的靖擧. 灌过比较可以看街. 本文方法的结某锝到的 、 SSJM及CIEDE2000值与其它六种方法相比都是最优的, 说明了本文方法的有效性.39 2 计導机攀报: _1苹0.50最逅* 为了验证本文方法所提出的浙进式双网络模型及双向约東损失函数的有效性, 我们对本文构建的两络做了太量前实验进行对比? 图S 给出f双浙进网络框架的几种变形网祭所得结桌的客观指标PSMR和SSfM的变化趋势? 其中, 纵坐标为客观指标懷,. 横坐标为网讚训练的迭代次数. 图中黑色粗线条显: 示的是本文渐进式双网络模型的M果值;黑色虚线条显示的是没有舉用'欢向绮束损朱得到的结粜值; M色点划幾条为本文方法塞础上加深网络深度〖 精细部分采用6 个卷视层结构 到的结果值; 黑色细线条是只有第一部分的增强模块得到的结果眞? 从?(a) 和(b)中指标曲线的走勢可以看出s只有第一部分增強模块得到的结果计算的PSM?和怒JM值都比较低; 其它三种网络结构包含去躁0.850.800. 75§0.70泛0.650.600.55模块的渐进式增强网络得到的增强结果的PSNJ?和SSIM值都有了明, 显的提高|但从黑色点划线条的走势来看, 增加精细化部分网络的深度相比釆用四个卷积层降低了客观指标值j从黑色虛钱条和黑色粗线条的走勢可以看迅, 采用本文提出的双向约束损失函数对两络的训. 练进一歩提高了两络的锖确性, 得封了吏高的揩标. 由此可以说明, 本文提:出的网络结构是有效的,具有较强鲁棒性-4 总结与展望本文提出了一种渐进式双网络结构的低曝光?像增强方法针对低曝光图像的亮. 度降质和噪声降质的坷题* 采: 用渐进式的思想先对图像进行堯废和色彩的增强, 然后再去除被放大的噪声, 针对这两: 步f喿作?举文输纖了菌个渐翁式 模: 輿: 来拟合现实场景亮度*暗到盡傭亮度变化. 及从粗到细的图僳恢复过. 隹, 使绾皋更自. 然? 为了更圩地训练网络^本文提出一种双向约束的损失函数, 不仅能疋向逼近真实结果, 还能反向迤近输入, 达到动态平衡, 提高网_确象为了验证所提出的方法的有效性, 本文做了大薰前实验,并与多个方法从主观和客观苘方面迸行比较r实验结果也验证了本文方法具宥更优, 的性能.未来我们计划将视频监控斑趙与图#增莪箅法研究相结合f 提窗实时视频监控在夜间或者光照不足等情况下的监控能力?图6[1]RaoY,ChenL.Asurveyofvideoenhancementtechniques.JournalofInformationHidingandMultimediaSignalProcessing,2012,13(1):71-9921参考 文 献30r25■20-15■10-PSNR\ll(a) 各方法的PSNR指标比较SSIMtCIEDE20Q0i(b) 各方法的SSIM指标比较 (c) 各方法的CIEDE2000指标比较图5 本文方法与其它方法的PSM?、SSJM以及CJEDE2000 指标比较(其中, PSM?和SSJM的值越大图像质量越好, CJEDE2000 越小图像质量越好)?M-L-DM-M-N15913172125293337414549535761656973778185899397epoch(a)SSIM指标结果图M-- -DMM-L-^MN(159131721252933374145 49535761656973778185899397epoch(b)PSNR指标结果图对比实验结果(M为本文方法, DM为在本文方法基础上加深网络深度, M-L为本文方法不使用双向约束损失函数, M-N为本文方只含增强模块)65432222225ATCOQH黄淑英等: 基于渐进式双网络模型的低曝光图像增强方法 39 32 期[2]LiL, WangL, WangW, GaoW. Alowlight imageenhancement met hodforbot hdenoisingandcont rastenlargi ng//ProceedingsoftheIEEEInternat ionalConferenceonImageProcessing. QuebecCity, Canada,2015: 37303734[3]LinS, WongC, RahmanM, etal. Image enhancement usi ngtheaveraginghi stogramequalizati on( AVI IEQ)approachf orcont rast improvementandbright nesspreservat ion. Computers&-Elect ricalEngineeri ng,2015 , 46: 356 370[4]SujeeR?Padmavat hiS. Image enhancement t hrough pyramidhistogrammat chi ng//Proceedingsof theIEEEInt ernat ionalConf erenceonComput erCommunicat ionandInformat ics.Coimbat ore, India,2017 : 1 5[5]SunSY,WangLP,ZhangBM,JingZL. Lowlightlevelimageenhancementbasedondual ist icsubimageandt wodimensionalhistogramanalysis. JournalofInfraredandMilli meterWaves,2003 ,22( 3) : 220224[6]JobsonDJ? RahmanZU, Woodel lGA. Propert iesandperf ormanceofacenter/surroundRet inex. IEEETransact ionsonImageProcessi ng,1997,6( 3) : 451 462[7]ZhangQing,YuanGan Zhao, XiaoChunXia,et al. Highqualityexposurecorrect ionofunderexposedphot os//Proceedingsof the ACMMult imediaConferenceonMultimediaConference.Seoul, Korea, 2018: 582 590[8]FuXue Yang, LiaoYingI Iao , ZengDe Lu, et al. Aprobabilist icmethodforimageenhancementwithsi mult aneousilluminationandref lect anceest imation. IEEETransactionsonImageProcessi ng, 2015 , 24( 12): 4 965 4977[9]LiMa Ding, LiuJi a Ying, YangWen I I an, GuoZongMing. Jointdenoisi ngandenhancementforlowlight imagesviaRet inexmodel. Int ernat ionalForumonDigit alTVandWirelessMultimediaCommunicat ions? 2017: 9199[10]LoreKG, Akint ayoA, SarkarS. LLNet: Adeepautoencoderapproachtonat urallowlightimageenhancement .Pat t ernRecognition,2017 , 61: 650 662[11]ZhangYongI I ua,ZhangJia Wan, GuoXi ao Jie. Kindli ngt hedarkness: Apracti callowlightI mageEnhancer//Proceedingsoft heACMInternat ionalConferenceonMul timedia. Nice,France,2019: 1632 1640[12]TaoL, ZhuC, XiangGQ, etal. LLCNN: Aconvol ut ionalneuralnetworkforlowlightimageenhancement/ /ProceedingsoftheIEEEVisualCommunicationsandImageProcessing.Florida , USA, 20 17: 1 4[13]ShenL, YueZI I , FangF, etal. MSRnet: Lowlight imageenhancement usingdeepconvolut ionalnet work. arXivpreprintarXiv: 1711. 0 248 8,2017[14]LandEI I . TheRet inext heoryofcolorvision. Scient ifi cAmerican,1977,237(6) : 108 129[15]JobsonDJ, RahmanZU, WoodellGA. AmultiscaleRetinexforbridgi ngt hegapbet weencolorimagesandt hehumanobservat ionof scenes. IEEETransact ionsonI mageProcessing,1997, 6(7) ; 965 976[16]FuX, ZengD, I l uangY, etal. Afusionbasedenhancingmet hodforweaklyilluminatedimages. SignalProcess, 201 6,12 9: 82 96[17]YingZ, LiG, GaoW. Abioinspiredmult i exposurefusionframeworkforlowlightimageenhancement . arXivpreprintarXiv: 1711. 00591 , 2017[18]GuoX, LiY, LingI I. LI ME: Lo wlightimageenhancementviail luminat ionmapest imation. IEEETransactionsonI mageProcessing,2017 ,26( 2) : 98 2 993[19]ChenChen,ChenQi Feng? Kolt unV. Learningtoseeinthedark//Proceedingsoft heIEEEConf erenceonComput erVi sionandPat ternRecognit ion. Salt LakeCity, Utah,2018:329133 00[20]WeiC, WangWJ, YangWI I , LiuJY. DeepRetinexdecomposi tio nf orlowl ight enhancement//Proceedingsof t heBri tishMachineVisionConference. Newcastle,UK,2018:1 10[21]JiangY, GongX, Li uD, etal. Enlight enGAN: Deeplightenhancementwithoutpairedsupervision. arXivpreprintarXiv: 1906. 06972 ,2019[22]WangR, ZhangQ, FuC, etal. Underexposedphotoenhancementusingdeepilluminationestimation//Proceedingsof theIEEEConferenceonComputerVisionandPat ternRecognit ion,LongBeach,USA,20 19: 6849 68 57[23]WangZ, Simoncel liEP, BovikAC. Multiscalest ruct uralsimilarityforimagequalityassessment//Proceedi ngsof the37 t hAsilomarConferenceonSignals?Syst emsandComputers.Pacifi cGrove,USA,2003 : 1398 1402[24]DongX, WangG,PangY,etal. Fastef fici ent algorithmforenhancementof lowlightingvideo//Proceedi ngsof theIEEEInt ernationalConferenceonMultimediaandExpo. Barcelona?Spain,2011: 1 6[25]DangNguyenDT, Pasqui niC, Conot t erV, BoatoG.RAISE: Arawimagesdatasetf ordigitalimagef orensics//ProceedingsoftheACMMultimediaSyst emsConference.Oregon, USA, 2015: 219 224[26]WangZ, BovikAC, SheikhI IR, Si moncelliEP. I magequalityassessment: Fromerrorvisibilitytost ruct uralsimilarity. IEEETransactionsonI mageProcessi ng, 2014 ,13 ( 4) : 600 612[27]SharmaG, WuWC, DalaiEN. TheCIEDE2000col ordifferenceformula : I mplement at ionnotes? supplementarytestdat a, andmat hemati calobservat ions. ColorResearchandApplicat ions,2005, 30 (1) : 213 039 4 计 算机 学 报 2021年HUANGShu-Ying,Ph.D.,associateprofessor.Hermainresearchinterestsincl udeimageprocessingandmachinel earning.HUWei,M.S.candidate.Hi sresearchinterestsincludeimagedehazingandimagerestoration.YANGYong, Ph.D. ,professor.Hismainresearchi nterestsi ncl udeimageprocessi ngandmachinel earning.LIHong-Xia, M.S.candi date.Herresearchinterestsincludedeeplearningandimagerestoration.WANGBi n,M.S.candidate.Hisresearchinterestsi ncl udedeeplearni ngandimagerestoration.BackgroundWiththedevelopmentofartificialintel ligenceandtheexcell entperformanceofneuralnetworki nthefieldofimageprocessing,themethodbasedondeeplearninghasbeenappl iedtovariousfieldsofimageprocessing.Suchasimagefusion,imagedefogging,imageenhancement,imagesuper?resol utionandimagequal ityevaluation.Experi mental resul tsshowthatdeeplearningiseffectiveforfeatureextractionandanalysisofdigitalimages.Withtherapidincreaseofthenumberandtypesofi magedata,thegeneralizationabilityofdeeplearningmethodhasbeengreatlychall enged.Therearethreekindsofimageenhancementmethods ,whicharebasedonhistogramequalization,Retinextheoryanddeepl earning.Histogrambasedmethodsmainlyenhancethevisualeffectbyenhancingtheimagecontrast.Suchmethodshavelowcomputationalcompl exity,butarepronetodetaillossandcolordeviationprobl ems.Retinexbasedmethodscangetthebettervisualeffect ,buttheystil lhavel imitationsinmodelsolving.Deeplearningbasedmethodscanavoidthel imitationsofmodelsolving,buttheyalsoignoretheimagingprincipl eofimages.Inthispaper ,wecombinetheadvantagesofdeeplearningandphysicalmodel ,andproposealowexposurei mageenhancementmethodbasedontheprogressivedualnetworkmodel,whichisdividedintotwomodules:imageenhancementandimagedenoi sing.Wealsoproposeabidirectionalconstraintlossfunctiontomakethel earningresultofnetworkapproachtherealdatafromthepositiveandnegativedirections.Theexperimentalresultsshowthatthemethodproposedinthispapercanachi eveexcell entresul ts. |
[返回] |