一种基于Retinex的暗图像增强方法技术

一种基于Retinex的暗图像增强方法，将图像从RGB图像转换成HSV图像，更加符合人眼视觉。将所有层直接连接在一起。在这种新型架构中，每层的输入由所有之前层的特征映射组成，其输出将传输给每个后续层。这些特征映射通过深度级联聚合。提升了神经网络的深度，解决了梯度消失问题。原始网络在网络末端用最近邻的方法上采样达到高分辨率尺寸，与原始网络相比改进EnhanceNet网络采用双线性插值上采样方法，采用这种方法不需要进行学习，运行速度快，操作简单。只需要设置好固定的参数值即可，设置的参数就是中心值需要乘以的系数。并且与原有最近邻算法上采样相比较采用新的上采样方法可以将图像显示更加清晰，网络效果进一步提升。升。升。

【技术实现步骤摘要】
一种基于Retinex的暗图像增强方法


[0001]本专利技术涉及图像处理
，具体涉及一种基于Retinex的暗图像增强方法。

技术介绍

[0002]随着科技进步，新的图像技术在逐渐推广，在日常生活中人们对于图像的要求也越来越高，针对阴天或夜晚等弱光条件下拍摄的图像具有亮度低、对比度低和细节模糊等问题，这个时候就需要把目标的亮度提高一点。传统基本做法是对于频域与空域进行处理，这些基本方法都只针对于特定场景下图像增强，对于复杂环境的图像增强效果并不明显，近年来，基于深度学习的图像增强技术取得卓越进展，在高级图像理解任务，比如：图像分类、目标检测等方面取得令人瞩目的成绩。基于深度学习算法的图像增强技术成为研究的热点，常规做法是将图像数据训练集暗图像与亮图像进行直接训练，得到权重参数但是修复完成后视觉效果以及图像质量有待提高。因此技术革新的新型暗图像增强技术是十分有必要的。

技术实现思路

[0003]本专利技术为了克服以上技术的不足，提供了一种使得暗图像增强，提升图像的细节以及对比度，将图像还原成光照充分的条件下的图像，提升图像的辨识度的暗图像增强方法。
[0004]本专利技术克服其技术问题所采用的技术方案是：
[0005]一种基于Retinex的暗图像增强方法，包括如下步骤：
[0006]a)将输入计算机的彩色图片进行颜色通道转换，将图片由RGB图像转换到HSV图像，得到H通道图像、S通道图像和V通道图像；
[0007]b)将H通道图像、S通道图像和V通道图像输入EnhanceNet网络，EnhanceNet网络为3*3卷积核，EnhanceNet网络对各个通道图像进行L次卷积处理，提取相关特征数据，每次卷积操作得到一层，得到L个层，使用ReLU及3*3的卷积得到深度为64的特征映射图像；
[0008]c)将DenseNet残差网络引入步骤b)中的EnhanceNet网络，将步骤b)中的L层使用DenseNet残差网络连接在一起，每个层的输入由所有之前层的特征映射组成；
[0009]d)将特征映射图像进行双线性插值上采样，采样后进行两次卷积操作；
[0010]e)使用ReLU和3*3卷积核对两次卷积操作后的特征图像进行卷积，得到输出结果I
est
，通过公式计算得到损失函数L
E
，其中I
HR
为原始高分辨率图像，通过计算损失函数得到优化后的图像；
[0011]f)将优化后的图像的H通道、S通道和V通道使用颜色通道合成RGB图像L(x,y)；
[0012]g)通过公式r(x,y)＝logR(x,y)＝logS(x,y)L(x,y)计算得到最终图像r(x,y)，R(x,y)为反射图像，将图像L(x,y)与原始图像进行求解卷积得到优化后的增强图像。
[0013]进一步的，步骤b)中的EnhanceNet网络采用步长为2的3*3的卷积核对H通道图像、
S通道图像和V通道图像进行卷积处理。
[0014]进一步的，步骤d)中使用ReLU和3*3卷积核进行卷积操作，卷积长度为特征映射图像深度的2倍，卷积宽度为特征映射图像深度的2倍。
[0015]进一步的，步骤d)中通过公式
[0016][0017]计算特征映射图像的双线性插值上采样获取的值f(x,y)，其中Q
11
、Q
12
、Q
21
及Q
22
为点(x,y)周边的任意四个点，Q
11
＝(x1,y1)，Q
12
＝(x1,y2)，Q
21
＝(x2,y1)，Q
22
＝(x2,y2)。
[0018]本专利技术的有益效果是：相比较传统暗图像增强算法，只是将原始图像与标注后的图像放入神经网络中进行训练，新技术将图像从RGB图像转换成HSV图像，更加符合人眼视觉。与原始网络相比较将DenseNet改进残差网络引入。将所有层直接连接在一起。在这种新型架构中，每层的输入由所有之前层的特征映射组成，其输出将传输给每个后续层。这些特征映射通过深度级联聚合。提升了神经网络的深度，解决了梯度消失问题。原始网络在网络末端用最近邻的方法上采样达到高分辨率尺寸，与原始网络相比改进EnhanceNet网络采用双线性插值上采样方法，采用这种方法不需要进行学习，运行速度快，操作简单。只需要设置好固定的参数值即可，设置的参数就是中心值需要乘以的系数。并且与原有最近邻算法上采样相比较采用新的上采样方法可以将图像显示更加清晰，网络效果进一步提升。
附图说明
[0019]图1为本专利技术的原理框图。
具体实施方式
[0020]下面结合附图1对本专利技术做进一步说明。
[0021]一种基于Retinex的暗图像增强方法，包括如下步骤：
[0022]a)将输入计算机的彩色图片进行颜色通道转换，将图片由RGB图像转换到HSV图像，得到H通道图像、S通道图像和V通道图像。
[0023]b)将H通道图像、S通道图像和V通道图像输入EnhanceNet网络，EnhanceNet网络为3*3卷积核，EnhanceNet网络对各个通道图像进行L次卷积处理，提取相关特征数据，每次卷积操作得到一层，得到L个层，使用ReLU及3*3的卷积得到深度为64的特征映射图像。这些特征映射通过深度级联聚合。提升了神经网络的深度，解决了梯度消失问题。
[0024]c)将DenseNet残差网络引入步骤b)中的EnhanceNet网络，将步骤b)中的L层使用DenseNet残差网络连接在一起，每个层的输入由所有之前层的特征映射组成。引入改进残差网络DenseNet，原始EnhanceNet网络中，假设有L层网络，但是在DenseNet中，就会有L(L+1)/2个连接，可以理解为每一层输入都来自前面所有层的输出。DenseNet原理公式如下：
[0025]X1＝H1([x0,x1,
…
,x
l-1
])，其中[x0,x1,
…
,x
l-1
]表示将0到l-1层的输出feature map做concatenation。concatenation是做通道的合并，H1包括BN、ReLU和3*3的卷积。
[0026]建立建立Block叠加，Block之间任意两层间建立前向Skip Connection是整个核心思想。一个Block包含多个Dense layer，其又包括三层：BN、ReLU和Conv。
[0027]前3层作为一个部分是为了利用1*1Conv减少特征图数量，增加计算量。不同dense block之间的连接层称为transition layer，每个transition layer包含3层：BN，1*1Conv和2*2AvgPool。
[0028]d)将特征映射图像进行双线性插值上采样，采样后进行两次卷积操作。采用这种方法不本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于Retinex的暗图像增强方法，其特征在于，包括如下步骤：a)将输入计算机的彩色图片进行颜色通道转换，将图片由RGB图像转换到HSV图像，得到H通道图像、S通道图像和V通道图像；b)将H通道图像、S通道图像和V通道图像输入EnhanceNet网络，EnhanceNet网络为3*3卷积核，EnhanceNet网络对各个通道图像进行L次卷积处理，提取相关特征数据，每次卷积操作得到一层，得到L个层，使用ReLU及3*3的卷积得到深度为64的特征映射图像；c)将DenseNet残差网络引入步骤b)中的EnhanceNet网络，将步骤b)中的L层使用DenseNet残差网络连接在一起，每个层的输入由所有之前层的特征映射组成；d)将特征映射图像进行双线性插值上采样，采样后进行两次卷积操作；e)使用ReLU和3*3卷积核对两次卷积操作后的特征图像进行卷积，得到输出结果I
est
，通过公式计算得到损失函数L
E
，其中I
HR
为原始高分辨率图像，通过计算损失函数得到优化后的图像；f)将优化后的图像的H通道、S通道和V通道使用颜色通道合成RGB图像L(x,y)；g)通过公式r(x,y)＝logR...

【专利技术属性】
技术研发人员：凌泽乐，高明，金长新，
申请(专利权)人：济南浪潮高新科技投资发展有限公司，
类型：发明
国别省市：