一种改进的Retinex-Net渐晕图像校正方法

本发明涉及数字图像处理，具体为一种改进的retinex-net渐晕图像校正方法。

背景技术：

1、在进行计算机视觉任务中，高质量的图片起着至关重要的作用，相机捕获的图像在处理不当时往往会发生退化，渐晕现象就是图像退化的一种表现，图像渐晕主要分为内部渐晕和物理渐晕，物理渐晕表现为在图像角落中突然出现强烈的变暗，其主要原因是在拍摄图片时镜头前面有大面积的遮挡或者错误使用遮光罩，内部渐晕通常表现为从图像中心到四角略微变暗，这种渐晕是由镜头或者机身内部引起的，一般情况下变焦镜头容易产生这种渐晕，不同程度渐晕的存在会使图片边缘丢失许多细节，模糊边缘信息，色彩出现较大偏差，渐晕现象在工业相机采集图像时是无法避免的，

2、近年来通过国内外学者的不断研究，目前已经提出几种校正图像渐晕的方法，通常，这些方法分为：基于直方图均衡化的渐晕校正处理算法、基于伽马校正的渐晕图像增强处理等，但是这些校正方法都存在着一些的弊端：约束条件或者是先验知识不够完善而引起噪声的增强、曝光过大，色彩失真、噪声增多等，无法达到令人满意的效果，

3、在渐晕图像校正领域，传统的校正算法要进行人工调参，不能适应于不同场景采集的渐晕图像，每当更换不同场景的图像之后，算法就要重新计算参数，由此就导致了校正算法的准确性和灵活性大幅度降低，随着机器学习的发展，卷积神经网络的提出很大程度上解决了图像处理领域存在的难题。

技术实现思路

1、(一)解决的技术问题

2、针对现有技术的不足，本发明提供了一种改进的retinex-net渐晕图像校正方法，解决了上述背景技术中所提出的问题。

3、(二)技术方案

4、本发明为了实现上述目的具体采用以下技术方案：

5、一种改进的retinex-net渐晕图像校正方法，包括以下步骤，

6、步骤1：图像预处理，批处理所采集图像集的尺寸、格式；

7、步骤2：在已训练好的改进的retinex-net模型中输入成对的渐晕和对照图像；

8、步骤3：将步骤2中输入的渐晕图像和对照图像通过分解网络，得到对照图像分解后的光照分量inormal和反射分量rnormal，渐晕图像分解后的光照分量ivignetting和反射分量rvignetting。将ivignetting、rvignetting输入到校正网络中分别进行增强和去噪处理；

9、步骤4：校正网络为多分支网络，在步骤3中输入的ivignetting通过校正网络进行增强处理得到光照分量的增强图像ienhance，rvignetting通过密集残差(residual dense block，rdb)去噪网络得到去噪处理后反射分量的增强图像renhance；

10、步骤5：将步骤4中处理过后的两个分量图像renhance和ienhance进行融合重建，获得校正之后的恢复图像。

11、进一步地，所述步骤1和2中将所采集的1000对渐晕图像和对照图像批处理为大小为21.17cm×14.11cm、格式为png的rgb图像，将其输入到改进的retinex-net模型中进行训练。

12、进一步地，所述步骤3中，分解模块的网络结构为：输入层中输入的成对的渐晕图像和对照图像，二者共享权重，利用两个约束条件将两幅图片分解为相对应的光照分量图像和反射分量图像，具体流程为:在隐藏层中，第一层通过卷积先进行特征提取，再通过两层卷积+relu激活层，利用两幅图像的约束函数得到两幅图像的光照分量图像inormal和ivignetting和反射分量图像rnormal和rvignetting，在输出层中，提取出的多通道特征通过大小为3x 3的卷积层中投影出光照分量和反射分量，再使用sigmoid函数进行约束变量，保证阈值为正。

13、进一步地，所述步骤4中，在校正网络中，包含的两个模块；

14、校正网络中的第一个模块为增强模块，在输入层中输入rvignetting图像和ivignetting图像，隐藏层中共五层，先通过两层卷积relu激活层，其目的是进行下采样操作，明确大尺度图像的光照分量情况，同时与镜像上采样层进行跳跃连接，重建图像局部光照分布，让网络强制学习残差，再通过一层大小为1x 1的卷积层将多尺度拼接后的多通道特征简化，最后通过一个3x 3的卷积输出层，重建光照分量图像；

15、校正网络中的第二个模块分空洞卷积网络模块，该网络结构是八层卷积，其目的是为了提高感受野，ivignetting图像并行经过空洞卷积网络和增强子网络，最后经过一个大小为1x 1的卷积层合并得到光照分量的增强图像ienhance。

16、进一步地，所述步骤4中，在反射去噪网络中，为了充分利用反射分量图像中的所有特征，提高网络性能，本发明在去噪网络中使用密集残差网络；

17、其rdb网络第一层是卷积+relu激活函数，经过卷积提取特征之后再通过relu激活形成非线性的特征映射，网络的第二层结构为残差块，其中包含三个dense block，每个dense block中又含有4层卷积+relu激活函数，每个卷积层获得模块中所有先前卷积层的输出，其中相邻的卷积采用短连接的方式，且所有卷积层跳跃连接在一起，第三层到第十一层的结构是卷积+relu激活+bn，即特征提取之后再进行特征映射，将特征进行规范化处理，最后的输出层为大小为3x 3的卷积层，得到训练后的去噪反射分量。

18、(三)有益效果

19、与现有技术相比，本发明提供了一种改进的retinex-net渐晕图像校正方法，具备以下有益效果：

20、本发明，结合retinex理论和卷积神经网络，利用约束条件将渐晕图像分解为光照分量图和反射分量图像，将光照分量图像增强处理，并通过空洞卷积网络扩大感受野，将反射分量图像输入密集残差网络对其进行去噪，再将处理后的两个分量融合重建，从而获得校正后的恢复图像。本发明算法与其他校正算法进行对比，具有优越的处理效果和处理性能，校正后的恢复图像十分接近原图，实用性与鲁棒性有很大提升。

技术特征：

1.一种改进的retinex-net渐晕图像校正方法，其特征在于：包括以下步骤，

2.根据权利要求1所述的一种改进的retinex-net渐晕图像校正方法，其特征在于：所述步骤1和2中将所采集的1000对渐晕图像和对照图像批处理为大小为21.17cm×14.11cm、格式为png的rgb图像，将其输入到改进的retinex-net模型中进行训练。

3.根据权利要求1所述的一种改进的retinex-net渐晕图像校正方法，其特征在于：所述步骤3中，分解模块的网络结构为：输入层中输入的成对的渐晕图像和对照图像，二者共享权重，利用两个约束条件将两幅图片分解为相对应的光照分量图像和反射分量图像，具体流程为:在隐藏层中，第一层通过卷积先进行特征提取，再通过两层卷积+relu激活层，利用两幅图像的约束函数得到两幅图像的光照分量图像inormal和ivignetting和反射分量图像rnormal和rvignetting，在输出层中，提取出的多通道特征通过大小为3x3的卷积层中投影出光照分量和反射分量，再使用sigmoid函数进行约束变量，保证阈值为正。

4.根据权利要求1所述的一种改进的retinex-net渐晕图像校正方法，其特征在于：所述步骤4中，在校正网络中，包含的两个模块；

5.根据权利要求1所述的一种改进的retinex-net渐晕图像校正方法，其特征在于：所述步骤4中，在反射去噪网络中，为了充分利用反射分量图像中的所有特征，提高网络性能，本发明在去噪网络中使用密集残差网络；

技术总结
本发明属于数字图像处理技术领域，尤其为一种改进的Retinex‑Net渐晕图像校正方法，包括以下步骤，步骤1：图像预处理，批处理所采集图像集的尺寸、格式；步骤2：在已训练好的改进的Retinex‑Net模型中输入成对的渐晕和对照图像。本发明结合Retinex理论和卷积神经网络，利用约束条件将渐晕图像分解为光照分量图和反射分量图像，将光照分量图像增强处理，并通过空洞卷积网络扩大感受野，将反射分量图像输入密集残差网络对其进行去噪，再将处理后的两个分量融合重建，从而获得校正后的恢复图像。本发明算法与其他校正算法进行对比，具有优越的处理效果和处理性能，校正后的恢复图像十分接近原图，实用性与鲁棒性有很大提升。

技术研发人员：黄丹丹,王菲,刘智,王惠绩,高晗
受保护的技术使用者：长春理工大学
技术研发日：
技术公布日：2024/1/14