使用 NumPy 对灰度图像进行直方图均衡化

Question

如何轻松地对存储在 NumPy 数组中的多个灰度图像进行直方图均衡？

我有这种 4D 格式的 96x96 像素 NumPy 数据：

 (1800, 1, 96,96)

原文由 pbu 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

Moose 指向此 博客条目 的 评论 很好地完成了这项工作。

为了完整起见，我在这里给出了一个示例，使用更好的变量名和在问题中的 4D 数组中的 1000 个 96x96 图像上循环执行。它很快（在我的电脑上 1-2 秒）并且只需要 NumPy。

 import numpy as np

def image_histogram_equalization(image, number_bins=256):
    # from http://www.janeriksolem.net/histogram-equalization-with-python-and.html

    # get image histogram
    image_histogram, bins = np.histogram(image.flatten(), number_bins, density=True)
    cdf = image_histogram.cumsum() # cumulative distribution function
    cdf = (number_bins-1) * cdf / cdf[-1] # normalize

    # use linear interpolation of cdf to find new pixel values
    image_equalized = np.interp(image.flatten(), bins[:-1], cdf)

    return image_equalized.reshape(image.shape), cdf

if __name__ == '__main__':

    # generate some test data with shape 1000, 1, 96, 96
    data = np.random.rand(1000, 1, 96, 96)

    # loop over them
    data_equalized = np.zeros(data.shape)
    for i in range(data.shape[0]):
        image = data[i, 0, :, :]
        data_equalized[i, 0, :, :] = image_histogram_equalization(image)[0]

原文由 Trilarion 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

非常快速和简单的方法是使用 skimage 模块提供的累积分布函数。基本上你用数学来证明它。

 from skimage import exposure
import numpy as np
def histogram_equalize(img):
    img = rgb2gray(img)
    img_cdf, bin_centers = exposure.cumulative_distribution(img)
    return np.interp(img, bin_centers, img_cdf)

原文由 Dodo 发布，翻译遵循 CC BY-SA 3.0 许可协议