一种颜色保持的特征融合去雾方法技术

本发明专利技术公开一种颜色保持的特征融合去雾方法，包括以下步骤：(1)获取总训练集和测试集；所述总训练集包括室外训练集和室内训练集，所述室外训练集、室内训练集和测试集内均由清晰图像和有雾图像组成；(2)基于卷积神经网络实现单幅图像去雾；卷积神经网络包括特征提取、特征融合和图像复原三部分；特征提取包括内容信息提取模块和风格信息提取模块，内容信息提取模块和风格信息提取模块分别采用七个级联的残差块RB1～RB7和三个级联的密集残差块RDB1～RDB3作为主体框架；特征融合部分通过残差块输出的特征图进行加权堆叠得到加权的内容特征图；图像复原部分包括一个卷积层，用于得到清晰的无雾图像。

【技术实现步骤摘要】
一种颜色保持的特征融合去雾方法
本专利技术属于计算机图像处理领域，特别是涉及一种颜色保持的特征融合去雾方法。
技术介绍
在雾霾等天气状况下，由于受大气悬浮颗粒、雾、霾等影响，户外采集到的图像出现对比度下降、颜色失真、细节信息丢失等，严重影响户外视觉系统发挥效用。因此，图像去雾在计算机视觉应用和数字图像处理领域尤为重要。目前，基于深度学习的方法成为单幅图像去雾的研究热点，该方法主要分为两类，第一类方法通过估计传输图和大气光值或其他之间变量来恢复无雾图像[1-3]，第二类方法通过直接学习有雾图像到无雾图像之间的映射关系来实现图像去雾[4-5]。上述方法虽然取得了较好的去雾效果，但是第一类方法依赖于大气散射模型，通过网络估计的传输图往往包含过多的细节信息，且大多数方法将大气光设置为全局一致常量，去雾后的图像易出现颜色失真现象。第二类端到端网络去雾方法容易造成部分区域去雾过度，导致去雾后图像出现过饱和现象。
技术实现思路
本专利技术的目的是为了克服现有技术中的不足，提供一种颜色保持的特征融合去雾方法。本专利技术提出的去雾网络本文档来自技高网...

【技术保护点】
1.一种颜色保持的特征融合去雾方法，其特征在于，包括以下步骤：/n(1)获取总训练集和测试集；所述总训练集包括室外训练集和室内训练集，所述室外训练集、室内训练集和测试集内均由清晰图像和有雾图像组成；/n(2)基于卷积神经网络实现单幅图像去雾；卷积神经网络包括特征提取、特征融合和图像复原三部分；特征提取包括内容信息提取模块和风格信息提取模块，内容信息提取模块和风格信息提取模块分别采用七个级联的残差块RB1～RB7和三个级联的密集残差块RDB1～RDB3作为主体框架；/n特征融合部分通过残差块输出的特征图进行加权堆叠得到加权的内容特征图；/n图像复原部分包括一个卷积层，用于得到清晰的无雾图像。/n

【技术特征摘要】
1.一种颜色保持的特征融合去雾方法，其特征在于，包括以下步骤：
(1)获取总训练集和测试集；所述总训练集包括室外训练集和室内训练集，所述室外训练集、室内训练集和测试集内均由清晰图像和有雾图像组成；
(2)基于卷积神经网络实现单幅图像去雾；卷积神经网络包括特征提取、特征融合和图像复原三部分；特征提取包括内容信息提取模块和风格信息提取模块，内容信息提取模块和风格信息提取模块分别采用七个级联的残差块RB1～RB7和三个级联的密集残差块RDB1～RDB3作为主体框架；
特征融合部分通过残差块输出的特征图进行加权堆叠得到加权的内容特征图；
图像复原部分包括一个卷积层，用于得到清晰的无雾图像。


2.根据权利要求1所述一种颜色保持的特征融合去雾方法，其特征在于，所述特征提取具体包括以下内容：
所述内容信息提取模块包含三个卷积层和七个残差块；三个卷积层分别为cov1，cov2，cov3，卷积核大小均为3*3，步长分别为1，1，2，padding均设置为1，输出通道数均为64；七个残差块分别是RB1，RB2，RB3，RB4，RB5，RB6，RB7；每个残差块的结构相同，均包含三个卷积层，卷积核的大小均为3*3，步长均为1，padding均设置为1，输出的通道数均为64；
所述风格信息提取模块包含两个卷积层和...

【专利技术属性】
技术研发人员：杨爱萍，刘瑾，王海新，何宇清，
申请(专利权)人：天津大学，
类型：发明
国别省市：天津;12