首页 > 其他分享> > 直方图均衡进行颜色校正

直方图均衡进行颜色校正

2022-07-11 09:03:08 作者：互联网

完整内容见参考资料；

本文只是对代码略微进行了改动；

原图是一张夜间拍摄的教堂照片，见下图

from skimage.exposure import histogram, cumulative_distribution
from skimage import img_as_ubyte
from skimage.color import rgb2gray
from scipy.stats import cauchy, logistic


def plot(gray, pltimg, title1='', title2=''):
    '''灰度图 画 直方图，pdf，cdf'''
    fig, ax = plt.subplots(1, 3, figsize=(15, 5))

    ## 手动计算后 画图
    ax0 = ax[0]
    ax0.imshow(pltimg)             # 图像
    ax0.set_title(title1)

    ax1 = ax[1]
    ax1.set_title(title2)
    ax11 = ax1.twinx()   # 二图合一，双坐标
    hist, bins = np.histogram(gray, bins=255)
    cumsum = np.cumsum(hist) / np.sum(hist)
    ax1.bar(bins[1:], hist / np.sum(hist), width=0.5)    # 概率密度函数
    ax11.plot(bins[1:], cumsum, 'r*-')                   # 累计概率分布

    ## skimage 计算后 画图
    ax2 = ax[2]
    ax22 = ax2.twinx()
    skhist_pdf, skbins_pdf = histogram(gray)
    skhist_cdf, skbins_cdf = cumulative_distribution(gray)
    ax2.bar(skbins_pdf, skhist_pdf / skhist_pdf.sum())
    ax22.plot(skbins_cdf, skhist_cdf, 'r*-')

    plt.show()

# 原始图像 - 灰度图
img = cv.imread('imgs/jiaotang.png')
gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
# 灰度图的直方图
plot(gray, img)

输出原图和灰度图的直方图

源图像明显偏暗，其直方图左侧明显占比过高，也证明了这点；

把原始直方图转换成均匀分布的直方图，对应的图像变成

# 灰度图 经过 直方图均衡化，转换成 均匀分布
image_intensity = img_as_ubyte(gray)
freq, bins = cumulative_distribution(image_intensity)

target_bins = np.arange(255)
target_freq = np.linspace(0, 1, len(target_bins))           # 一条直线，从0到1的累加
new_vals = np.interp(freq, target_freq, target_bins)        # 把原累加 插值成 上面的直线，x为 range(255)
gray2 = new_vals[image_intensity]
plot(gray2, gray2)

输出

图像整体亮度增加，因为把大量的低区间像素转换到高区间像素了；对比度也增强了；

均匀分布的累计概率分布就是一条直线

直方图均衡不仅可以转换成均匀分布，也可以转换成其他分布；

本文实现 柯西分布cauchy、逻辑分布logistic，其效果更优于均匀分布，仅对本图而言

def individual_channel(image, dist, channel):
    im_channel = img_as_ubyte(image[:, :, channel])
    freq, bins = cumulative_distribution(im_channel)
    new_vals = np.interp(freq, dist.cdf(np.arange(0, 256)),
                         np.arange(0, 256))
    return new_vals[im_channel].astype(np.uint8)


def distribution(image, function, mean, std):
    dist = function(mean, std)
    fig, ax = plt.subplots(1, 2, figsize=(15, 5))

    image_intensity = img_as_ubyte(rgb2gray(image))
    freq, bins = cumulative_distribution(image_intensity)
    ax[0].step(bins, freq, c='b', label='Actual CDF')
    ax[0].plot(dist.cdf(np.arange(0, 256)),
               c='r', label='Target CDF')
    ax[0].legend()
    ax[0].set_title('Actual vs. Target Cumulative Distribution')

    red = individual_channel(image, dist, 0)
    green = individual_channel(image, dist, 1)
    blue = individual_channel(image, dist, 2)
    ax[1].imshow(np.dstack((red, green, blue)))
    ax[1].set_title('Transformed Image')
    return ax

distribution(img, cauchy, 90, 30)
plt.show()

distribution(img, logistic, 90, 30)
plt.show()

输出

我们已经探索了如何使用直方图处理来校正图像中的颜色，实现了各种分布函数，以了解它如何影响结果图像中的颜色分布。

同样，我们可以得出结论，在固定图像的颜色强度方面没有“一体适用”的解决方案，数据科学家的主观决定是确定哪个是最适合他们的图像处理需求的解决方案

参考资料：

https://mp.weixin.qq.com/s/T1ziuMkMhqIc0NDHmgm19A　　使用直方图处理进行颜色校正

https://github.com/jephraim-manansala/histogram-manipulation　　github 代码

标签：校正,img,image,直方图,均衡,np,ax,bins
来源： https://www.cnblogs.com/yanshw/p/15570837.html

直方图均衡 进行 颜色校正

直方图均衡进行颜色校正