CN113344798A - 一种基于Retinex的暗图像增强方法 - Google Patents
一种基于Retinex的暗图像增强方法 Download PDFInfo
- Publication number
- CN113344798A CN113344798A CN202010101186.5A CN202010101186A CN113344798A CN 113344798 A CN113344798 A CN 113344798A CN 202010101186 A CN202010101186 A CN 202010101186A CN 113344798 A CN113344798 A CN 113344798A
- Authority
- CN
- China
- Prior art keywords
- image
- network
- channel
- convolution
- enhancenet
- 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
- 238000000034 method Methods 0.000 title claims abstract description 33
- 238000005070 sampling Methods 0.000 claims abstract description 6
- 238000013507 mapping Methods 0.000 claims description 13
- 238000012545 processing Methods 0.000 claims description 7
- 238000006243 chemical reaction Methods 0.000 claims description 3
- 230000002708 enhancing effect Effects 0.000 claims description 3
- 239000000284 extract Substances 0.000 claims description 3
- 230000002194 synthesizing effect Effects 0.000 claims description 3
- 238000013528 artificial neural network Methods 0.000 abstract description 6
- 230000000694 effects Effects 0.000 abstract description 5
- 230000008034 disappearance Effects 0.000 abstract description 4
- 230000006870 function Effects 0.000 description 8
- 238000005516 engineering process Methods 0.000 description 7
- 238000013135 deep learning Methods 0.000 description 3
- 238000012549 training Methods 0.000 description 3
- 230000007704 transition Effects 0.000 description 2
- 230000000007 visual effect Effects 0.000 description 2
- 230000009286 beneficial effect Effects 0.000 description 1
- 238000004364 calculation method Methods 0.000 description 1
- 238000007796 conventional method Methods 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 238000001514 detection method Methods 0.000 description 1
- 238000010586 diagram Methods 0.000 description 1
- 238000009499 grossing Methods 0.000 description 1
- 238000005286 illumination Methods 0.000 description 1
- 238000003707 image sharpening Methods 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 230000008569 process Effects 0.000 description 1
- 238000011160 research 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
- G06T5/90—Dynamic range modification of images or parts thereof
- G06T5/92—Dynamic range modification of images or parts thereof based on global image properties
-
- 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/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/90—Determination of colour characteristics
-
- 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
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (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)
- Computer Vision & Pattern Recognition (AREA)
- Image Processing (AREA)
Abstract
一种基于Retinex的暗图像增强方法,将图像从RGB图像转换成HSV图像,更加符合人眼视觉。将所有层直接连接在一起。在这种新型架构中,每层的输入由所有之前层的特征映射组成,其输出将传输给每个后续层。这些特征映射通过深度级联聚合。提升了神经网络的深度,解决了梯度消失问题。原始网络在网络末端用最近邻的方法上采样达到高分辨率尺寸,与原始网络相比改进EnhanceNet网络采用双线性插值上采样方法,采用这种方法不需要进行学习,运行速度快,操作简单。只需要设置好固定的参数值即可,设置的参数就是中心值需要乘以的系数。并且与原有最近邻算法上采样相比较采用新的上采样方法可以将图像显示更加清晰,网络效果进一步提升。
Description
技术领域
本发明涉及图像处理技术领域,具体涉及一种基于Retinex的暗图像增强方法。
背景技术
随着科技进步,新的图像技术在逐渐推广,在日常生活中人们对于图像的要求也越来越高,针对阴天或夜晚等弱光条件下拍摄的图像具有亮度低、对比度低和细节模糊等问题,这个时候就需要把目标的亮度提高一点。传统基本做法是对于频域与空域进行处理,这些基本方法都只针对于特定场景下图像增强,对于复杂环境的图像增强效果并不明显,近年来,基于深度学习的图像增强技术取得卓越进展,在高级图像理解任务,比如:图像分类、目标检测等方面取得令人瞩目的成绩。基于深度学习算法的图像增强技术成为研究的热点,常规做法是将图像数据训练集暗图像与亮图像进行直接训练,得到权重参数但是修复完成后视觉效果以及图像质量有待提高。因此技术革新的新型暗图像增强技术是十分有必要的。
发明内容
本发明为了克服以上技术的不足,提供了一种使得暗图像增强,提升图像的细节以及对比度,将图像还原成光照充分的条件下的图像,提升图像的辨识度的暗图像增强方法。
本发明克服其技术问题所采用的技术方案是:
一种基于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)将特征映射图像进行双线性插值上采样,采样后进行两次卷积操作;
f)将优化后的图像的H通道、S通道和V通道使用颜色通道合成RGB图像L(x,y);
g)通过公式r(x,y)=logR(x,y)=logS(x,y)L(x,y)计算得到最终图像r(x,y),R(x,y)为反射图像,将图像L(x,y)与原始图像进行求解卷积得到优化后的增强图像。
进一步的,步骤b)中的EnhanceNet网络采用步长为2的3*3的卷积核对H通道图像、S通道图像和V通道图像进行卷积处理。
进一步的,步骤d)中使用ReLU和3*3卷积核进行卷积操作,卷积长度为特征映射图像深度的2倍,卷积宽度为特征映射图像深度的2倍。
进一步的,步骤d)中通过公式
计算特征映射图像的双线性插值上采样获取的值f(x,y),其中Q11、Q12、Q21及Q22为点(x,y)周边的任意四个点,Q11=(x1,y1),Q12=(x1,y2),Q21=(x2,y1),Q22=(x2,y2)。
本发明的有益效果是:相比较传统暗图像增强算法,只是将原始图像与标注后的图像放入神经网络中进行训练,新技术将图像从RGB图像转换成HSV图像,更加符合人眼视觉。与原始网络相比较将DenseNet改进残差网络引入。将所有层直接连接在一起。在这种新型架构中,每层的输入由所有之前层的特征映射组成,其输出将传输给每个后续层。这些特征映射通过深度级联聚合。提升了神经网络的深度,解决了梯度消失问题。原始网络在网络末端用最近邻的方法上采样达到高分辨率尺寸,与原始网络相比改进EnhanceNet网络采用双线性插值上采样方法,采用这种方法不需要进行学习,运行速度快,操作简单。只需要设置好固定的参数值即可,设置的参数就是中心值需要乘以的系数。并且与原有最近邻算法上采样相比较采用新的上采样方法可以将图像显示更加清晰,网络效果进一步提升。
附图说明
图1为本发明的原理框图。
具体实施方式
下面结合附图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残差网络连接在一起,每个层的输入由所有之前层的特征映射组成。引入改进残差网络DenseNet,原始EnhanceNet网络中,假设有L层网络,但是在DenseNet中,就会有L(L+1)/2个连接,可以理解为每一层输入都来自前面所有层的输出。DenseNet原理公式如下:
X1=H1([x0,x1,…,xl-1]),其中[x0,x1,…,xl-1]表示将0到l-1层的输出featuremap做concatenation。concatenation是做通道的合并,H1包括BN、ReLU和3*3的卷积。
建立建立Block叠加,Block之间任意两层间建立前向Skip Connection是整个核心思想。一个Block包含多个Dense layer,其又包括三层:BN、ReLU和Conv。
前3层作为一个部分是为了利用1*1Conv减少特征图数量,增加计算量。不同denseblock之间的连接层称为transition layer,每个transition layer包含3层:BN,1*1Conv和2*2AvgPool。
d)将特征映射图像进行双线性插值上采样,采样后进行两次卷积操作。采用这种方法不需要进行学习,运行速度快,操作简单。只需要设置好固定的参数值即可,设置的参数就是中心值需要乘以的系数。并且与原有最近邻算法上采样相比较采用新的上采样方法可以将图像显示更加清晰,网络效果进一步提升。
e)在本网络中参数的初始化采用Xavier网络的学习目标是超分辨率图像与输入的低分辨率图像线性插值的差。其中损失函数部分我们选择pixel-wise loss强调的是两幅图像之间每个对应像素的匹配。通过pixel-wise loss训练的图片通常会较为平滑。即使输出图片具有较高的PSNR,视觉效果也并没有很突出。后期引入图像锐化以解决过于平滑问题。使用ReLU和3*3卷积核对两次卷积操作后的特征图像进行卷积,得到输出结果Iest,通过公式计算得到损失函数LE,其中IHR为原始高分辨率图像,通过计算损失函数得到优化后的图像。损失函数LE用来估量你模型的预测值f(x)与真实值Y的不一致程度,它是一个非负实值函数,损失函数越小,模型的鲁棒性就越好。
f)将优化后的图像的H通道、S通道和V通道使用颜色通道合成RGB图像L(x,y),L(x,y)即为模型处理的光照度图;
g)通过公式r(x,y)=logR(x,y)=logS(x,y)L(x,y)计算得到最终图像r(x,y),R(x,y)为反射图像,将图像L(x,y)与原始图像进行求解卷积得到优化后的增强图像。
相比较传统暗图像增强算法,只是将原始图像与标注后的图像放入神经网络中进行训练,新技术将图像从RGB图像转换成HSV图像,更加符合人眼视觉。与原始网络相比较将DenseNet改进残差网络引入。将所有层直接连接在一起。在这种新型架构中,每层的输入由所有之前层的特征映射组成,其输出将传输给每个后续层。这些特征映射通过深度级联聚合。提升了神经网络的深度,解决了梯度消失问题。原始网络在网络末端用最近邻的方法上采样达到高分辨率尺寸,与原始网络相比改进EnhanceNet网络采用双线性插值上采样方法,采用这种方法不需要进行学习,运行速度快,操作简单。只需要设置好固定的参数值即可,设置的参数就是中心值需要乘以的系数。并且与原有最近邻算法上采样相比较采用新的上采样方法可以将图像显示更加清晰,网络效果进一步提升。基于Retinex原理将图像分成为反射图像S与亮度图像L,原理中讲解L与原始图像存在较弱相关性,创新的使用现在深度学习技术,寻找亮度图像与原图像中的关系,进而计算出之间插值,而不是直接采用回归的做法将网络与之进行拟合。
优选的,步骤b)中的EnhanceNet网络采用步长为2的3*3的卷积核对H通道图像、S通道图像和V通道图像进行卷积处理,最终获得深度h为64的特征映射图像。
优选的,步骤d)中使用ReLU和3*3卷积核进行卷积操作,卷积长度为特征映射图像深度的2倍,卷积宽度为特征映射图像深度的2倍。即2w*2h*64,其中w为图像宽度,h为图像长度。
优选的,步骤d)中通过公式
计算特征映射图像的双线性插值上采样获取的值f(x,y),其中Q11、Q12、Q21及Q22为点(x,y)周边的任意四个点,Q11=(x1,y1),Q12=(x1,y2),Q21=(x2,y1),Q22=(x2,y2)。
以上所述仅为本发明的较佳实施例,仅用于说明本发明的技术方案,并非用于限定本发明的保护范围。凡在本发明的精神和原则之内所做的任何修改、等同替换、改进等,均包含在本发明的保护范围内。
Claims (4)
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)将特征映射图像进行双线性插值上采样,采样后进行两次卷积操作;
f)将优化后的图像的H通道、S通道和V通道使用颜色通道合成RGB图像L(x,y);
g)通过公式r(x,y)=logR(x,y)=logS(x,y)L(x,y)计算得到最终图像r(x,y),R(x,y)为反射图像,将图像L(x,y)与原始图像进行求解卷积得到优化后的增强图像。
2.根据权利要求1所述的基于Retinex的暗图像增强方法,其特征在于:步骤b)中的EnhanceNet网络采用步长为2的3*3的卷积核对H通道图像、S通道图像和V通道图像进行卷积处理。
3.根据权利要求1所述的基于Retinex的暗图像增强方法,其特征在于:步骤d)中使用ReLU和3*3卷积核进行卷积操作,卷积长度为特征映射图像深度的2倍,卷积宽度为特征映射图像深度的2倍。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010101186.5A CN113344798A (zh) | 2020-02-18 | 2020-02-18 | 一种基于Retinex的暗图像增强方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010101186.5A CN113344798A (zh) | 2020-02-18 | 2020-02-18 | 一种基于Retinex的暗图像增强方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN113344798A true CN113344798A (zh) | 2021-09-03 |
Family
ID=77467105
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202010101186.5A Pending CN113344798A (zh) | 2020-02-18 | 2020-02-18 | 一种基于Retinex的暗图像增强方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN113344798A (zh) |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110084108A (zh) * | 2019-03-19 | 2019-08-02 | 华东计算技术研究所(中国电子科技集团公司第三十二研究所) | 基于gan神经网络的行人重识别系统及方法 |
CN110136060A (zh) * | 2019-04-24 | 2019-08-16 | 西安电子科技大学 | 基于浅层密集连接网络的图像超分辨率重建方法 |
CN110796599A (zh) * | 2019-10-15 | 2020-02-14 | 哈尔滨工程大学 | 视网膜图像超分辨率重建的通道加权生成式对抗网络方法 |
-
2020
- 2020-02-18 CN CN202010101186.5A patent/CN113344798A/zh active Pending
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN110084108A (zh) * | 2019-03-19 | 2019-08-02 | 华东计算技术研究所(中国电子科技集团公司第三十二研究所) | 基于gan神经网络的行人重识别系统及方法 |
CN110136060A (zh) * | 2019-04-24 | 2019-08-16 | 西安电子科技大学 | 基于浅层密集连接网络的图像超分辨率重建方法 |
CN110796599A (zh) * | 2019-10-15 | 2020-02-14 | 哈尔滨工程大学 | 视网膜图像超分辨率重建的通道加权生成式对抗网络方法 |
Non-Patent Citations (4)
Title |
---|
GAO HUANG ET AL.: "Densely Connected Convolutional Networks", 《ARXIV:1608.06993V5 [CS.CV]》 * |
MEHDI S. M. SAJJADI ET AL.: "EnhanceNet: Single Image Super-Resolution Through Automated Texture Synthesis", 《ARXIV:1612.07919V2 [CS.CV]》 * |
TONG TONG ET AL.: "Image Super-Resolution Using Dense Skip Connections", 《2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV)》 * |
肖亮 等: "《基于图像先验建模的超分辨增强理论与算法 变分PDE、稀疏正则化与贝叶斯方法》", 31 July 2017 * |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN108921786B (zh) | 基于残差卷积神经网络的图像超分辨率重构方法 | |
CN110136060B (zh) | 基于浅层密集连接网络的图像超分辨率重建方法 | |
CN112001843B (zh) | 一种基于深度学习的红外图像超分辨率重建方法 | |
CN112435191A (zh) | 一种基于多个神经网络结构融合的低照度图像增强方法 | |
CN112801904B (zh) | 一种基于卷积神经网络的混合退化图像增强方法 | |
CN116152120B (zh) | 一种融合高低频特征信息的低光图像增强方法及装置 | |
CN113284061B (zh) | 一种基于梯度网络的水下图像增强方法 | |
CN112508812A (zh) | 图像色偏校正方法、模型训练方法、装置及设备 | |
CN112561791A (zh) | 一种基于优化AnimeGAN的图像风格迁移 | |
CN115018708A (zh) | 一种基于多尺度特征融合的机载遥感图像超分辨率重建方法 | |
CN111340696A (zh) | 融合仿生视觉机制的卷积神经网络图像超分辨率重建方法 | |
CN116486074A (zh) | 一种基于局部和全局上下文信息编码的医学图像分割方法 | |
CN113409355A (zh) | 一种基于fpga的运动目标识别系统及方法 | |
CN115953311A (zh) | 基于Transformer的多尺度特征表示的图像去雾方法 | |
CN116934592A (zh) | 一种基于深度学习的图像拼接方法、系统、设备及介质 | |
CN114565539A (zh) | 一种基于线上知识蒸馏的图像去雾方法 | |
Cai et al. | Underwater image processing system for image enhancement and restoration | |
CN108492264B (zh) | 一种基于sigmoid变换的单帧图像快速超分辨方法 | |
CN114926359B (zh) | 联合双颜色空间恢复和多级解码结构的水下图像增强方法 | |
CN117351340A (zh) | 基于双颜色空间的水下图像增强算法 | |
CN110415816B (zh) | 一种基于迁移学习的皮肤病临床图像多分类方法 | |
CN113344798A (zh) | 一种基于Retinex的暗图像增强方法 | |
US11997246B2 (en) | Trained artificial intelligence model for raw to RGB image transformation | |
CN111612709B (zh) | 一种基于DnCNNs改进的图像降噪方法 | |
CN114511462B (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 | ||
RJ01 | Rejection of invention patent application after publication | ||
RJ01 | Rejection of invention patent application after publication |
Application publication date: 20210903 |