首页 / 算法 / 自动色彩均衡（ACE）快速算法

自动色彩均衡（ACE）快速算法

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了自动色彩均衡（ACE）快速算法，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2795字，纯文字阅读大概需要4分钟。

内容图文

    ACE算法源自retinex算法，可以调整图像的对比度，实现人眼色彩恒常性和亮度恒常性，通过差分来计算目标点与周围像素点的相对明暗关系来校正最终像素值，有很好的增强效果。但是计算复杂度非常高，本文提出一种有效的快速实现方法。

    为叙述方便，这里假设后面的图像都是归一化到[0,1]之间的浮点数图像。

    ACE算法的计算公式为：

Y =   ∑(g(I(x0)-I(x))w(x0,x))  / ∑(w(x0,x))                                    x属于I                                (1)

其中，w是权重参数，离中心点像素越远w值越小，可以直接取值欧氏距离。g()是相对对比度调节参数，非线性的，简单取如下计算方法：

g(x) = max(min(ax, 1.0), -1.0)                                                                                        （2）

这里a是控制参数，值越大，细节增强越明显。计算完后，还要对Y进行一次归一化即可得到最终的增强图像。

    ACE的增强效果普遍与retinex好。需要注意的是，ACE中当前像素是与整个图像的其他像素做差分比较，计算复杂度非常非常高，这也是限制它应用的最主要原因，本文主要基于两个假设：（1）对一副图像ACE增强后得到输出Y，如果对Y再进行一次ACE增强，输出仍然是Y本身；（2）对一副图像的ACE增强结果进行尺寸缩放得到Y，对Y进行ACE增强，输出仍然是Y本身。这两个假设我没法证实，呵呵，就算臆想的吧。

    如果上面假设成立，我们就可以对图像进行缩放得到I1，对I1的ACE增强结果进行尺度放大（与I尺寸一样）得到Y1，那么Y和Y1是非常接近的，我们只需要在Y1基础上进一步处理即可。这里就又引申了两个细节问题：1）如何快速的求I1的ACE增强结果？ 其实很简单，对它再次缩放得到I2，求I2的增强结果，依次类推，就是金字塔结构思想。2)如何在Y1基础上进一步处理得到Y？ 因为是在整个图像域进行差分比较运算，与近处邻域像素的比较构成了Y的细节信息，与远处像素的比较构成了Y的全局背景信息，那么我们合理假设，Y和Y1的全局背景信息相同，只更新细节信息即可，也就是，我们需要在Y1基础上加上I中邻近像素的差分结果，并减去Y1中邻近像素的差分结果就是最终的输出Y。

    上面说的有点绕，慢慢看吧。

    下面是python代码

import cv2
import numpy as np
import math

def stretchImage(data, s=0.005, bins = 2000):    #线性拉伸，去掉最大最小0.5%的像素值，然后线性拉伸至[0,1]
    ht = np.histogram(data, bins);
    d = np.cumsum(ht[0])/float(data.size)
    lmin = 0; lmax=bins-1
    while lmin<bins:
        if d[lmin]>=s:
            break
        lmin+=1
    while lmax>=0:
        if d[lmax]<=1-s:
            break
        lmax-=1
    return np.clip((data-ht[1][lmin])/(ht[1][lmax]-ht[1][lmin]), 0,1)

g_para = {}
def getPara(radius = 5):                        #根据半径计算权重参数矩阵
    global g_para
    m = g_para.get(radius, None)
    if m is not None:
        return m
    size = radius*2+1
    m = np.zeros((size, size))
    for h in range(-radius, radius+1):
        for w in range(-radius, radius+1):
            if h==0 and w==0:
                continue
            m[radius+h, radius+w] = 1.0/math.sqrt(h**2+w**2)
    m /= m.sum()
    g_para[radius] = m
    return m

def zmIce(I, ratio=4, radius=300):                     #常规的ACE实现
    para = getPara(radius)
    height,width = I.shape
    zh,zw = [0]*radius + range(height) + [height-1]*radius, [0]*radius + range(width)  + [width -1]*radius
    Z = I[np.ix_(zh, zw)]
    res = np.zeros(I.shape)
    for h in range(radius*2+1):
        for w in range(radius*2+1):
            if para[h][w] == 0:
                continue
            res += (para[h][w] * np.clip((I-Z[h:h+height, w:w+width])*ratio, -1, 1))
    return res

def zmIceFast(I, ratio, radius):                #单通道ACE快速增强实现
    height, width = I.shape[:2]
    if min(height, width) <=2:
        return np.zeros(I.shape)+0.5
    Rs = cv2.resize(I, ((width+1)/2, (height+1)/2))
    Rf = zmIceFast(Rs, ratio, radius)             #递归调用
    Rf = cv2.resize(Rf, (width, height))
    Rs = cv2.resize(Rs, (width, height))

    return Rf+zmIce(I,ratio, radius)-zmIce(Rs,ratio,radius)    
            
def zmIceColor(I, ratio=4, radius=3):               #rgb三通道分别增强，ratio是对比度增强因子，radius是卷积模板半径
    res = np.zeros(I.shape)
    for k in range(3):
        res[:,:,k] = stretchImage(zmIceFast(I[:,:,k], ratio, radius))
    return res

if __name__ == ‘__main__‘:
    m = zmIceColor(cv2.imread(‘p4.bmp‘)/255.0)*255
    cv2.imwrite(‘zmIce.jpg‘, m)

下面是实验结果,上边是原图，下边是增强结果。

 

 技术分享技术分享

 技术分享技术分享

技术分享 技术分享

技术分享 技术分享

 

原文：http://www.cnblogs.com/zmshy2128/p/6135551.html

内容总结

以上是互联网集市为您收集整理的自动色彩均衡（ACE）快速算法全部内容，希望文章能够帮你解决自动色彩均衡（ACE）快速算法所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

扫描二维码推送至手机访问。

本文链接：https://qyyshop.com/info/1199175.html

来源：【匿名】