基于直方图的图像增强算法(HE、CLAHE、Retinex)

作为图像增强算法系列的第二篇文章,下面我们将要介绍功能强大、用途广泛、影响深远的对比度有限的自适应直方图均衡(CLAHE,Contrast Limited Adaptive Histogram Equalization)算法。尽管最初它仅仅是被当作一种图像增强算法被提出,但是现今在图像去雾、低照度图像增强,水下图像效果调节、以及数码照片改善等方面都有应用。这个算法的算法原理看似简单,但是实现起来却并不那么容易。我们将结合相应的Matlab代码来对其进行解释。希望你在阅读本文之前对朴素的直方图均衡算法有所了解,相关内容可以参见本系列的前一篇文章

先来看一下待处理的图像效果:


基于直方图的图像增强算法(HE、CLAHE、Retinex)_第1张图片


下面是利用CLAHE算法处理之后得到的两个效果(后面我们还会具体介绍我们所使用的策略):


基于直方图的图像增强算法(HE、CLAHE、Retinex)_第2张图片    基于直方图的图像增强算法(HE、CLAHE、Retinex)_第3张图片

                             效果图A                                                                         效果图B

对于一幅图像而言,它不同区域的对比度可能差别很大。可能有些地方很明亮,而有些地方又很暗淡。如果采用单一的直方图来对其进行调整显然并不是最好的选择。于是人们基于分块处理的思想提出了自适应的直方图均衡算法AHE。维基百科上说的也比较明白:AHE improves on this by transforming each pixel with a transformation function derived from a neighbourhood region. 但是这种方法有时候又会将一些噪声放大,这是我们所不希望看到的。于是荷兰乌得勒支大学的Zuiderveld教授又引入了CLAHE,利用一个对比度阈值来去除噪声的影响。特别地,为了提升计算速度以及去除分块处理所导致的块边缘过渡不平衡效应,他又建议采用双线性插值的方法。关于算法的介绍和描述,下面这两个资源已经讲得比较清楚。

事实上,尽管这个算法原理,然而它实现起来却仍然有很多障碍。但在此之前,笔者还需说明的是,Matlab中已经集成了实现CLAHE的函数adapthisteq(),如果你仅仅需要一个结果,其实直接使用这个函数就是最好的选择。我给出一些示例代码用以生成前面给出之效果。函数adapthisteq()只能用来处理灰度图,如果要处理彩色图像,则需要结合自己编写的代码来完成。上一篇文章介绍了对彩色图像进行直方图均衡的两种主要策略:一种是对R、G、B三个通道分别进行处理;另外一种是转换到另外一个色彩空间中再进行处理,例如HSV(转换后只需对V通道进行处理即可)。

首先,我们给出对R、G、B三个通道分别使用adapthisteq()函数进行处理的示例代码:

[plain]  view plain  copy
 
  1. img = imread('space.jpg');  
  2. rimg = img(:,:,1);  
  3. gimg = img(:,:,2);  
  4. bimg = img(:,:,3);  
  5. resultr = adapthisteq(rimg);  
  6. resultg = adapthisteq(gimg);  
  7. resultb = adapthisteq(bimg);  
  8. result = cat(3, resultr, resultg, resultb);  
  9. imshow(result);  
上述程序之结果效果图A所示。
  • 故事

    动物故事

    成语故事

    益智故事

    寓言故事

    童话故事

    民间故事

    历史故事

    四字成语故事

    中外民间故事

    世界民间故事

    安徒生童话故事

    格林童话故事

    365夜童话故事

    伊索寓言故事

    三国成语故事

    http://www.zaojuzi.com/zhfupo/15113.html

    http://www.zaojuzi.com/zhfupo/15112.html

    http://www.zaojuzi.com/zhfupo/15111.html

    http://www.zaojuzi.com/zhfupo/15110.html

    http://www.zaojuzi.com/zhfupo/15109.html

    http://www.zaojuzi.com/zhfupo/15108.html

    http://www.zaojuzi.com/zhfupo/15107.html

    http://www.zaojuzi.com/zhfupo/15106.html

    http://www.zaojuzi.com/zhfupo/15105.html

    http://www.zaojuzi.com/zhfupo/15104.html

    http://www.zaojuzi.com/zhfupo/15103.html

    http://www.zaojuzi.com/zhfupo/15102.html

    http://www.zaojuzi.com/zhfupo/15101.html

    http://www.zaojuzi.com/zhfupo/15100.html

    http://www.zaojuzi.com/zhfupo/15099.html

    http://www.zaojuzi.com/zhfupo/15098.html

    http://www.zaojuzi.com/zhfupo/15097.html

    http://www.zaojuzi.com/zhfupo/15096.html

    http://www.zaojuzi.com/zhfupo/15095.html

    http://www.zaojuzi.com/zhfupo/15094.html

    http://www.zaojuzi.com/zhfupo/15093.html

    http://www.zaojuzi.com/zhfupo/15092.html

    http://www.zaojuzi.com/zhfupo/15091.html

    http://www.zaojuzi.com/zhfupo/15090.html

    http://www.zaojuzi.com/zhfupo/15089.html

    http://www.zaojuzi.com/zhfupo/15088.html

    http://www.zaojuzi.com/zhfupo/15087.html

    http://www.zaojuzi.com/zhfupo/15086.html

    http://www.zaojuzi.com/zhfupo/15085.html

    http://www.zaojuzi.com/zhfupo/15084.html

    http://www.zaojuzi.com/zhfupo/15083.html

    http://www.zaojuzi.com/zhfupo/15082.html

    http://www.zaojuzi.com/zhfupo/15081.html

    http://www.zaojuzi.com/zhfupo/15080.html

    http://www.zaojuzi.com/zhfupo/15079.html

    http://www.zaojuzi.com/zhfupo/15078.html

    http://www.zaojuzi.com/zhfupo/15077.html

    http://www.zaojuzi.com/zhfupo/15076.html

    http://www.zaojuzi.com/zhfupo/15075.html

    http://www.zaojuzi.com/zhfupo/15074.html

    http://www.zaojuzi.com/zhfupo/13314.html

    http://www.zaojuzi.com/zhfupo/13313.html

    http://www.zaojuzi.com/zhfupo/13312.html

    http://www.zaojuzi.com/zhfupo/13310.html

    http://www.zaojuzi.com/zhfupo/13309.html

    http://www.zaojuzi.com/zhfupo/13308.html

    http://www.zaojuzi.com/zhfupo/13307.html

    http://www.zaojuzi.com/zhfupo/13306.html

    http://www.zaojuzi.com/zhfupo/13305.html

    http://www.zaojuzi.com/zhfupo/13303.html

    http://www.zaojuzi.com/zhfupo/13302.html

    http://www.zaojuzi.com/zhfupo/13301.html

    http://www.zaojuzi.com/zhfupo/13300.html

    http://www.zaojuzi.com/zhfupo/13299.html

    http://www.zaojuzi.com/zhfupo/13298.html

    http://www.zaojuzi.com/zhfupo/13297.html

    http://www.zaojuzi.com/zhfupo/13296.html

    http://www.zaojuzi.com/zhfupo/13295.html

    http://www.zaojuzi.com/zhfupo/13294.html

    http://www.zaojuzi.com/zhfupo/13293.html

    http://www.zaojuzi.com/zhfupo/13292.html

    http://www.zaojuzi.com/zhfupo/13291.html

    http://www.zaojuzi.com/zhfupo/13290.html

    http://www.zaojuzi.com/zhfupo/13289.html

    http://www.zaojuzi.com/zhfupo/13288.html

    http://www.zaojuzi.com/zhfupo/13287.html

    http://www.zaojuzi.com/zhfupo/13286.html

    http://www.zaojuzi.com/zhfupo/13279.html

    http://www.zaojuzi.com/zhfupo/12285.html

    http://www.zaojuzi.com/zhfupo/page_2.html

    http://www.zaojuzi.com/zhfupo/page_4.html

    http://www.zaojuzi.com/zhfupo/page_5.html

    下面程序将原图像的色彩空间转换到LAB空间之后再对L通道进行处理。

    [plain]  view plain  copy
     
    1. clear;  
    2. img = imread('space.jpg');  
    3. cform2lab = makecform('srgb2lab');  
    4. LAB = applycform(img, cform2lab);  
    5. L = LAB(:,:,1);  
    6. LAB(:,:,1) = adapthisteq(L);  
    7. cform2srgb = makecform('lab2srgb');  
    8. J = applycform(LAB, cform2srgb);  
    9. imshow(J);  

    上述程序所得之结果如图B所示。


    如果你希望把这个算法进一步提升和推广,利用用于图像去雾、低照度图像改善和水下图像处理,那么仅仅知其然是显然不够的,你还必须知其所以然。希望我下面一步一步实现的代码能够帮你解开这方面的困惑。鉴于前面所列之文献已经给出了比较详细的算法描述,下面将不再重复这部分内容,转而采用Matlab代码来对其中的一些细节进行演示。


    首先来从灰度图的CLAHE处理开始我们的讨论。为此清理一下Matlab的环境然后,读入一张图片(并将其转化灰度图),获取图片的长、宽、像素灰度的最大值、最小值等信息。

    [plain]  view plain  copy
     
    1. clc;  
    2. clear all;  
    3. Img = rgb2gray(imread('space.jpg'));  
    4. [h,w] = size(Img);  
    5. minV = double(min(min(Img)));  
    6. maxV = double(max(max(Img)));  
    7. imshow(Img);  

    图像的初始状态显示如下。此外该图的 Height = 395,Width = 590,灰度最大值为255,最小值为8。

    基于直方图的图像增强算法(HE、CLAHE、Retinex)_第4张图片


    我们希望把原图像水平方向分成8份,把垂直方向分成4份,即原图将被划分成4 × 8 = 32个SubImage。然后可以算得每个块(tile)的height = 99,width = 74。注意,由于原图的长、宽不太可能刚好可被整除,所以我在这里的处理方式是建立一个稍微大一点的图像,它的宽和长都被补上了deltax和deltay,以保证长、宽都能被整除。

    [plain]  view plain  copy
     
    1. NrX = 8;  
    2. NrY = 4;  
    3. HSize = ceil(h/NrY);  
    4. WSize = ceil(w/NrX);  
    5.   
    6. deltay = NrY*HSize - h;  
    7. deltax = NrX*WSize - w;  
    8.   
    9. tmpImg = zeros(h+deltay,w+deltax);  
    10. tmpImg(1:h,1:w) = Img;  

    对长和宽进行填补之后,对新图像的一些必要信息进行更新。


    [plain]  view plain  copy
     
    1. new_w = w + deltax;  
    2. new_h = h + deltay;  
    3.   
    4. NrPixels = WSize * WSize;  

    然后指定图像中直方图横坐标上取值的计数(也就指定了统计直方图上横轴数值的间隔或计数的精度),对于色彩比较丰富的图像,我们一般都要求这个值应该大于128。

    [cpp]  view plain  copy
     
    1. % NrBins - Number of greybins for histogram ("dynamic range")  
    2. NrBins = 256;  


    然后用原图的灰度取值范围重新映射了一张Look-Up Table(当然你也可以直接使用0~255这个范围,这取决你后续建立直方图的具体方法),并以此为基础为每个图像块(tile)建立直方图。

    [cpp]  view plain  copy
     
    1. LUT = zeros(maxV+1,1);  
    2.   
    3. for i=minV:maxV  
    4.     LUT(i+1) = fix(i - minV);%i+1  
    5. end  
    6.   
    7. Bin = zeros(new_h, new_w);  
    8. for m = 1 : new_h  
    9.     for n = 1 : new_w  
    10.         Bin(m,n) = 1 + LUT(tmpImg(m,n) + 1);  
    11.     end  
    12. end  
    13.   
    14. Hist = zeros(NrY, NrX, 256);  
    15. for i=1:NrY  
    16.     for j=1:NrX  
    17.         tmp = uint8(Bin(1+(i-1)*HSize:i*HSize, 1+(j-1)*WSize:j*WSize));  
    18.         %tmp = tmpImg(1+(i-1)*HSize:i*HSize,1+(j-1)*WSize:j*WSize);  
    19.         [Hist(i, j, :), x] = imhist(tmp, 256);  
    20.     end  
    21. end  
    22.   
    23. Hist = circshift(Hist,[0, 0, -1]);  

    注意:按通常的理解,上面这一步我们应该建立的直方图(集合)应该是一个4×8=32个长度为256的向量(你当然也可以这么做)。但由于涉及到后续的一些处理方式,我这里是生成了一个长度为256的4×8矩阵。Index = 1的矩阵其实相当于是整张图像各个tile上灰度值=0的像素个数计数。例如,我们所得的Hist(:, :, 18)如下。这就表明图像中最左上角的那个tile里面灰度值=17的像素有零个。同理,它右边的一个tile则有46个灰度值=17的像素。


    [plain]  view plain  copy
     
    1. Hist(:,:,18) =  
    2.   
    3.      0    46   218    50    14    55    15     7  
    4.      0     0    21    18   114    15    74    73  
    5.      0     1     0     0     2    67   124    82  
    6.      0     0     0     0     0     1     9     2  

    然后来对直方图进行裁剪。Matlab中内置的函数adapthisteq()中cliplimit参数的取值范围是0~1。这里我们所写的方法则要求该值>1。当然这完全取决你算法实现的策略,它们本质上并没有差异。然后我们将得到新的(裁剪后的)映射直方图。

    [plain]  view plain  copy
     
    1. ClipLimit = 2.5;  
    2. ClipLimit = max(1,ClipLimit * HSize * WSize/NrBins);  
    3.   
    4. Hist = clipHistogram(Hist,NrBins,ClipLimit,NrY,NrX);  
    5. Map=mapHistogram(Hist, minV, maxV, NrBins, NrPixels, NrY, NrX);  

    最后,也是最关键的步骤,我们需要对结果进程插值处理。这也是Zuiderveld设计的算法中最复杂的部分。

    [plain]  view plain  copy
     
    1. yI = 1;  
    2. for i = 1:NrY+1  
    3.     if i == 1  
    4.         subY = floor(HSize/2);  
    5.         yU = 1;  
    6.         yB = 1;  
    7.     elseif i == NrY+1  
    8.         subY = floor(HSize/2);  
    9.         yU = NrY;  
    10.         yB = NrY;  
    11.     else  
    12.         subY = HSize;  
    13.         yU = i - 1;  
    14.         yB = i;  
    15.     end  
    16.     xI = 1;  
    17.     for j = 1:NrX+1  
    18.         if j == 1  
    19.             subX = floor(WSize/2);  
    20.             xL = 1;  
    21.             xR = 1;  
    22.         elseif j == NrX+1  
    23.             subX = floor(WSize/2);  
    24.             xL = NrX;  
    25.             xR = NrX;  
    26.         else  
    27.             subX = WSize;  
    28.             xL = j - 1;  
    29.             xR = j;  
    30.         end  
    31.         UL = Map(yU,xL,:);  
    32.         UR = Map(yU,xR,:);  
    33.         BL = Map(yB,xL,:);  
    34.         BR = Map(yB,xR,:);  
    35.         subImage = Bin(yI:yI+subY-1,xI:xI+subX-1);  
    36.   
    37.         %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%  
    38.         sImage = zeros(size(subImage));  
    39.         num = subY * subX;  
    40.         for i = 0:subY - 1  
    41.             inverseI = subY - i;  
    42.             for j = 0:subX - 1  
    43.                 inverseJ = subX - j;  
    44.                 val = subImage(i+1,j+1);  
    45.                 sImage(i+1, j+1) = (inverseI*(inverseJ*UL(val) + j*UR(val)) ...  
    46.                                 + i*(inverseJ*BL(val) + j*BR(val)))/num;  
    47.             end  
    48.         end  
    49.         %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%  
    50.           
    51.         output(yI:yI+subY-1, xI:xI+subX-1) = sImage;  
    52.         xI = xI + subX;  
    53.     end  
    54.     yI = yI + subY;  
    55. end  

    这个地方,作者原文中已经讲得比较清楚了,我感觉我也没有必要狗尾续貂,班门弄斧了。下面截作者原文中的一段描述,足以说明问题。

    基于直方图的图像增强算法(HE、CLAHE、Retinex)_第5张图片

    最后来看看我们处理的效果如何(当然,这里还需要把之前我们填补的部分裁掉)。


    [plain]  view plain  copy
     
    1. output = output(1:h, 1:w);  
    2. figure, imshow(output, []);  

    来看看结果吧~可以对比一下之前的灰度图,不难发现,图像质量已有大幅改善。

    基于直方图的图像增强算法(HE、CLAHE、Retinex)_第6张图片


    ===========从来不看博客私信的主页君说:通常我会把我所了解的算法尽可能讲清楚,但除了必要的辅助演示之代码,我不会再额外提供源码下载,无论什么时候人都应该靠自己,只有当你能真正写出代码时,那个代码才是属于你的。===========无冥冥之志者,无昭昭之明,无惛惛之事者,无赫赫之功。===========如果你是图像处理的同道中人,欢迎加入图像处理学习群(529549320)。为避免广告推销等骚扰信息,入群需回答门槛问题(例如:x平方的一阶导数等于多少?有木有感觉so easy,不过不要笑,管理员收到的答案是五花八门的!注意每人仅有一次尝试机会哦)。Cheers~

  • 你可能感兴趣的:(基于直方图的图像增强算法(HE、CLAHE、Retinex))