新技能 Get，使用直方图处理进行颜色校正

作者 | 小白

来源 | 小白学视觉

在这篇文章中，我们将探讨如何使用直方图处理技术来校正图像中的颜色。

像往常一样，我们导入库，如numpy和matplotlib。此外，我们还从skimage 和scipy.stats库中导入特定函数。

import numpy as np
import matplotlib.pyplot as plt
from skimage.io import imread, imshow
from skimage import img_as_ubyte
from skimage.color import rgb2gray
from skimage.exposure import histogram, cumulative_distribution
from scipy.stats import cauchy, logistic

让我们使用马尼拉内穆罗斯马尼拉大教堂的夜间图像。

cathedral = imread('cathedral.jpg')
plt.imshow(cathedral)
plt.title('Manila Cathedral')

首先，让我们将图像转换为灰度。

fig, ax = plt.subplots(1,2, figsize=(15,5))
cathedral_gray = rgb2gray(cathedral)
ax[0].imshow(cathedral_gray, cmap='gray')
ax[0].set_title('Grayscale Image')
ax1 = ax[1]
ax2 = ax1.twinx()
freq_h, bins_h = histogram(cathedral_gray)
freq_c, bins_c = cumulative_distribution(cathedral_gray)
ax1.step(bins_h, freq_h*1.0/freq_h.sum(), c='b', label='PDF')
ax2.step(bins_c, freq_c, c='r',  label='CDF')
ax1.set_ylabel('PDF', color='b')
ax2.set_ylabel('CDF', color='r')
ax[1].set_xlabel('Intensity value')
ax[1].set_title('Histogram of Pixel Intensity');

由于图像是在夜间拍摄的，因此图像的特征比较模糊，这也在像素强度值的直方图上观察到，其中 PDF 在较低的光谱上偏斜。

由于图像的强度值是倾斜的，因此可以应用直方图处理来重新分布图像的强度值。直方图处理的目的是将图像的实际 CDF 拉伸到新的目标 CDF 中。通过这样做，倾斜到较低光谱的强度值将转换为较高的强度值，从而使图像变亮。

让我们尝试在灰度图像上实现这一点，我们假设 PDF 是均匀分布，CDF 是线性分布。

image_intensity = img_as_ubyte(cathedral_gray)
freq, bins = cumulative_distribution(image_intensity)
target_bins = np.arange(255)
target_freq = np.linspace(0, 1, len(target_bins))
new_vals = np.interp(freq, target_freq, target_bins)
fig, ax = plt.subplots(1,2, figsize=(15,5))
ax[0].step(bins, freq, c='b', label='Actual CDF')
ax[0].plot(target_bins, target_freq, c='r', label='Target CDF')
ax[0].legend()
ax[0].set_title('Grayscale: Actual vs. '
                'Target Cumulative Distribution')
ax[1].imshow(new_vals[image_intensity].astype(np.uint8), 
             cmap='gray')
ax[1].set_title('Corrected Image in Grayscale');

通过将实际 CDF 转换为目标 CDF，我们可以在保持图像关键特征的同时使图像变亮。请注意，这与仅应用亮度过滤器完全不同，因为亮度过滤器只是将图像中所有像素的强度值增加相等的量。在直方图处理中，像素强度值可以根据目标 CDF 增加或减少。

现在，让我们尝试在彩色图像中实现直方图处理。这些过程可以从灰度图像中复制——然而，不同之处在于我们需要对图像的每个通道应用直方图处理。为了简化实现，我们创建一个函数来对图像执行此过程。

def show_linear_cdf(image, channel, name, ax):
    image_intensity = img_as_ubyte(image[:,:,channel])
    freq, bins = cumulative_distribution(image_intensity)
    target_bins = np.arange(255)
    target_freq = np.linspace(0, 1, len(target_bins))
    ax.step(bins, freq, c='b', label='Actual CDF')
    ax.plot(target_bins, target_freq, c='r', label='Target CDF')
    ax.legend()
    ax.set_title('{} Channel: Actual vs. '
                 'Target Cumulative Distribution'.format(name))
def linear_distribution(image, channel):
    image_intensity = img_as_ubyte(image[:,:,channel])
    freq, bins = cumulative_distribution(image_intensity)
    target_bins = np.arange(255)
    target_freq = np.linspace(0, 1, len(target_bins))
    new_vals = np.interp(freq, target_freq, target_bins)
    return new_vals[image_intensity].astype(np.uint8)

现在，我们将这些函数应用于原始图像的每个通道。

fig, ax = plt.subplots(3,2, figsize=(12,14))
red_channel = linear_distribution(cathedral, 0)
green_channel = linear_distribution(cathedral, 1)
blue_channel = linear_distribution(cathedral, 2)
show_linear_cdf(cathedral, 0, ‘Red’, ax[0,0])
ax[0,1].imshow(red_channel, cmap=’Reds’)
ax[0,1].set_title(‘Corrected Image in Red Channel’)
show_linear_cdf(cathedral, 1, ‘Green’, ax[1,0])
ax[1,1].imshow(green_channel, cmap=’Greens’)
ax[1,1].set_title(‘Corrected Image in Green Channel’)
show_linear_cdf(cathedral, 2, ‘Blue’, ax[2,0])
ax[2,1].imshow(blue_channel, cmap=’Blues’)
ax[2,1].set_title(‘Corrected Image in Blue Channel’)

请注意，所有通道几乎都具有相同的 CDF，这显示了图像中颜色的良好分布——只是颜色集中在较低的强度值光谱上。就像我们在灰度图像中所做的一样，我们还将每个通道的实际 CDF 转换为目标 CDF。

校正每个通道的直方图后，我们需要使用 numpy stack函数将这些通道堆叠在一起。请注意，RGB 通道在堆叠时需要按顺序排列。

fig, ax = plt.subplots(1,2, figsize=(15,5))
ax[0].imshow(cathedral);
ax[0].set_title('Original Image')
ax[1].imshow(np.dstack([red_channel, green_channel, blue_channel]));
ax[1].set_title('Transformed Image');

堆叠所有通道后，我们可以看到转换后的图像颜色与原始图像的显着差异。直方图处理最有趣的地方在于，图像的不同部分会有不同程度的像素强度转换。请注意，马尼拉大教堂墙壁的像素强度发生了巨大变化，而马尼拉大教堂钟楼的像素强度却保持相对不变。

现在，让我们尝试使用其他函数作为目标 CDF 来改进这一点。为此，我们将使用该scipy.stats库导入各种分布，还创建了一个函数来简化我们的分析。

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

让我们使用 Cauchy 函数来试试这个。

distribution(cathedral, cauchy, 90, 30);

使用不同的分布似乎会产生更令人愉悦的配色方案。事实上，大教堂正门的弧线在逻辑分布中比线性分布更好，这是因为在逻辑分布中像素值强度的平移比线性分布要小，这可以从实际 CDF 线到目标 CDF 线的距离看出。

让我们看看我们是否可以使用逻辑分布进一步改进这一点。

distribution(cathedral, logistic, 90, 30);

请注意，门中的灯光如何从线性和Cauchy分布改进为逻辑分布的。这是因为逻辑函数的上谱几乎与原始 CDF 一致。因此，图像中的所有暗物体（低像素强度值）都被平移，而灯光（高像素强度值）几乎保持不变。

结论

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

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

Github代码连接：

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

往

期

回

顾

技术

基于聚类的图像分割（Python）

资讯

人工智能监考VS传统方式监考

资讯

Meta研发触觉手套助力元宇宙

图像

深度学习视频理解之图像分类

分享

点收藏

点点赞

点在看