本发明涉及图像处理技术领域，尤其涉及一种基于retinex的彩色图像增强方法。
背景技术：
眼底图像的质量对于疾病的检测以及获得可靠的诊断结果具有至关重要的作用。但视网膜眼底图像受成像设备、成像环境的影响，所获图像质量效果差，图像亮度和对比度很低，这些因素使得直接通过拍摄的眼底图像进行病理分析有一定困难，且诊断容易产生偏差，因此需要提高此类眼底图像的亮度和对比度，增强细节部分，便于分析。因此，在临床上，眼底图像的增强对于心血管疾病的诊断以及治疗评价具有重要的意义。
目前常用的视网膜图像增强方法有直方图均衡化、对比度受限自适应直方图均衡化、数学形态学方法以及hessian矩阵增强等方法。直方图均衡化是应用比较多的方法，但是由于增强后的灰度级减少，图像的细节会消失。为了弥补此问题，又提出了对比度受限自适应直方图均衡化算法，但是这种方法对于灰度级比较集中的图像增强效果不明显。数学形态学滤波可以很好地提取图像亮暗区域以及细节部分，但主要针对灰度图像来进行提取。基于hessian矩阵的滤波方法，对实现视网膜图像的全局增强是不合适的。眼底图像增强方法一般是增强血管信息以使其更有利于下一步血管的提取，而很少能够实现对视网膜图像中的血管、视盘和病灶等有用信息进行有保留的增强；另一方面，已有的眼底图像增强方法大部分是将彩色图像转化为对灰度图形进行增强的，因此，很难实现图像视觉效果保真。在具体的增强方法中，我们需要多方面的考虑，不仅需要突出图像的某些细节部分，还要对图像的动态范围和色彩信息等有所考虑。
基于人类视觉系统的retinex图像增强算法有着传统增强算法不可比拟的颜色恒常效果，增强后的图像局部对比度显著提高，能够有效地提高图像暗区域内的细节部分且色彩信息可得到很好的恢复，对提高医生的临床准确性有很大帮助。最早的单尺度retinex(singlescaleretinex,ssr)算法，虽可提高图像的对比度，但不能兼顾动态范围压缩及色调恢复，且会出现光晕现象。为了克服ssr算法的缺点，rahman等提出了多尺度retinex(multi-scaleretinex,msr)算法，但是msr色彩失真较大，随后又提出了带色彩恢复的多尺度retinex(multi-scaleretinexwithcolorrestoration,msrcr)算法，实现了对msr处理结果的色彩补偿。但是msrcr在计算过程中需要进行大量的卷积运算，会降低算法运算效率，并且经过处理后的图像会超出显示器的显示范围，常规的处理方法容易引起彩色图像的失真。因此，对处理视网膜眼底彩色图像增强鲜见报道，主要原因是眼底图像的结构明暗对比度明显，动态范围比较大，采用msrcr方法处理后发现光晕现象明显，无法观察到视神经盘，并且细节部分丢失严重，图像颜色总体泛白化。
技术实现要素：
本发明旨在提供一种基于retinex的彩色图像增强方法，能够改善彩色图像的亮度和对比度，避免彩色图像的失真，消除了对比度明显区域容易产生光晕的问题，并能增强图像边缘信息，突出图像细节。
为达到上述目的，本发明采用的技术方案如下：
一种基于retinex的彩色图像增强方法，包括：
获取源rgb图像，并获取所述源rgb图像中的源r通道图像、源g通道图像和源b通道图像；计算所述源rgb图像的灰度值，得到第一灰度图像；对所述第一灰度图像做多尺度retinex处理，得到第一反射图像；对所述第一反射图像进行图像像素修正，得到第二灰度图像；根据所述第二灰度图像和所述第一灰度图像计算出第一染色因子；根据所述第一染色因子分别对所述源g通道图像和源b通道图像进行染色处理，得到第二g通道图像和第二b通道图像；对所述源r通道图像进行染色处理，得到第三r通道图像；将所述第三r通道图像、第二g通道图像和第二b通道图像进行图像合成，获得所述源rgb图像增强后的彩色图像。
优选地，所述对所述源r通道图像进行染色处理，得到第三r通道图像的方法包括：
对所述源r通道图像做多尺度retinex处理后，再进行图像像素修正，得到第二r通道图像；根据所述第二r通道图像和所述源r通道图像计算出第二染色因子；取所述第一染色因子和第二染色因子中的较大值，根据所述较大值对所述源r通道图像进行染色处理，得到第三r通道图像。
优选地，所述对所述第一反射图像进行图像像素修正，得到第二灰度图像的方法为：
其中，rmsr(x,y)为所述第一反射图像，
优选地，所述第一染色因子为所述第二灰度图像与所述第一灰度图像的比值；所述第二染色因子为所述第二r通道图像与所述源r通道图像的比值。
优选地，所述根据所述第一染色因子分别对所述源g通道图像和源b通道图像进行染色处理，得到第二g通道图像和第二b通道图像的方法为：
i′g(x,y)＝ig(x,y)×factor1
i′b(x,y)＝ib(x,y)×factor1
其中，factor1为第一染色因子，ig(x,y)为源g通道图像，ib(x,y)为源b通道图像，i′g(x,y)为第二g通道图像，i′b(x,y)为第二b通道图像。
本发明实施例提供的基于retinex的彩色图像增强方法，通过对源rgb图像的第一灰度图像进行多尺度retinex处理后，再进行图像像素的修正，根据修正后的第二灰度图像和第一灰度图像可获取第一染色因子，根据第一染色因子对源rgb图像的g、b两个通道图像分别进行染色，再对源rgb图像的r通道图像进行染色，将处理后的三个通道合成最终增强后的彩色图像。尤其对图像像素进行修正时，剪切了直方图两端的小概率值，使得更大范围的灰度级用于表示图像的细节，因此提高了图像对比度。本发明提供的技术方案，能够改善彩色图像的亮度和对比度，避免彩色图像的失真，消除了对比度明显区域容易产生光晕的问题，并能增强图像边缘信息，突出图像细节。
附图说明
图1为本发明实施例的方法流程图；
图2为本发明实施例中的原始眼底彩色图像；
图3为采用现有的msrcr算法对图2进行处理后的图像；
图4为采用本发明的方法对图2进行处理后的图像。
具体实施方式
为了使本发明的目的、技术方案及优点更加清楚明白，以下结合附图，对本发明进行进一步详细说明。
步骤101，获取源rgb图像，并获取所述源rgb图像中的源r通道图像、源g通道图像和源b通道图像；
本实施例中，将获取的源rgb图像先转换为double型，再分别提取源r通道图像信息、源g通道图像信息和源b通道图像信息，并分别记为r(x,y)、g(x,y)、b(x,y)。
步骤102，计算所述源rgb图像的灰度值，得到第一灰度图像i(x,y)；
本实施例中，根据以下公式计算源rgb图像的灰度值，该公式也是在hsi(hue，saturation，intensity)色彩空间提取亮度的方法：
i(x,y)＝(r(x,y)+g(x,y)+b(x,y))/3
式中的r(x,y)、g(x,y)、b(x,y)即为步骤101中的r(x,y)、g(x,y)、b(x,y)。
步骤103，对所述第一灰度图像i(x,y)做多尺度retinex处理，得到第一反射图像rmsr(x,y)：
式中，n为尺度数目,rmrs(x,y)表示多尺度的输出结果,即第一反射图像，wn为对应的每一个尺度的权值，g(x,y,c)表示高斯环绕函数，c为高斯函数的标准偏差，在本实施例中分别选择小尺度、中尺度、大尺度的高斯环绕函数。根据参数表设定三个尺度参数，这里我们取15,80,250三个尺度。k为归一化因子，使高斯环绕函数满足：
∫∫g(x,y,c)dxdy＝1
步骤104，对所述第一反射图像rmsr(x,y)进行图像像素修正，得到第二灰度图像；
由于上述处理为对数域的输出，可能会出现负值，也有可能会超出显示器的显示范围，因此必须要进行适当的修正将像素压缩到显示器可以显示的范围[0,255]内。关于这个映射的算法(增强补偿算法)，具有重要的意义，它的好坏直接决定了最后输出的图像质量。本实施例主要采用scb(simplestcolorbalance)进行增益补偿，公式描述如下所示：
其中，rmsr(x,y)为所述第一反射图像，
本实施例中我们把s1，s2都设置为1，使图像中的r、g、b每个通道上的值通过本步骤的公式变换映射到[0,255]之间，因此，剪切了第一反射图像的直方图中两端小概率值，使得更大范围的灰度级用于表示图像的细节，因此提高了图像对比度。
步骤105，根据所述第一灰度图像和所述第二灰度图像计算出第一染色因子factor1；
本实施例中，所述第一染色因子为所述第二灰度图像与所述第一灰度图像的比值，即
步骤106，根据所述第一染色因子factor1分别对所述源g通道图像和源b通道图像进行染色处理，得到第二g通道图像和第二b通道图像，具体方法为：
i′g(x,y)＝ig(x,y)×factor1
i′b(x,y)＝ib(x,y)×factor1
其中，factor1为第一染色因子，ig(x,y)为源g通道图像，ib(x,y)为源b通道图像，i′g(x,y)为第二g通道图像，i′b(x,y)为第二b通道图像。
步骤107，对所述源r通道图像进行染色处理，得到第三r通道图像；
由于r通道的亮度信息最多，经过亮度调整后(即染色处理后)信息可能存在丢失情况，若直接用第一染色因子对源r通道图像进行染色，得到的图像可能颜色偏红，因此需要对源r通道图像进行另外的染色处理，具体方法包括：
(1)对所述源r通道图像做多尺度retinex处理后，再进行图像像素修正，得到第二r通道图像；
其中，对源r通道图像做多尺度retinex处理的公式为：
式中，ir(x,y)为源r通道图像，其它参数意义与公式(1)中相同。
再根据步骤104中的公式(2)进行scb(simplestcolorbalance)的增益补偿(gain/offset)处理，即图像像素修正：
式中，
(2)根据所述第二r通道图像
(3)取所述第一染色因子factor1和第二染色因子factor2中的较大值，根据所述较大值对所述源r通道图像ir(x,y)进行染色处理，得到第三r通道图像i″r(x,y)。
步骤108，将所述第三r通道图像i″r(x,y)、第二g通道图像i′g(x,y)和第二b通道图像i′b(x,y)进行图像合成，获得所述源rgb图像增强后的彩色图像。
本发明实施例提供的基于retinex的彩色图像增强方法，通过对源rgb图像的第一灰度图像进行多尺度retinex处理后，再进行图像像素的修正，根据修正后的第二灰度图像和第一灰度图像可获取第一染色因子，根据第一染色因子对源rgb图像的g、b两个通道图像分别进行染色，再对源rgb图像的r通道图像进行染色，将处理后的三个通道合成最终增强后的彩色图像。尤其在对图像像素进行修正时，剪切了直方图两端的小概率值，使得更大范围的灰度级用于表示图像的细节，因此提高了图像对比度。本发明提供的方法可称为cp-msr(colorpreservebasedonmsr)，经过实验结果表明，该方法处理的视网膜眼底图像血管对比度提升、色彩保护较好以及图像细节清晰，避免了颜色失真，解决了明暗交替明显区域的光晕现象,因此，对临床上进行病理分析以及治疗评价有重要的意义。
以上所述，仅为本发明的具体实施方式，但本发明的保护范围并不局限于此，任何熟悉本技术领域的技术人员在本发明揭露的技术范围内，可轻易想到变化或替换，都应涵盖在本发明的保护范围之内。