匀色 直方图匹配python

在Python中实现直方图匹配可以使用OpenCV库。下面是一个简单的示例代码：

import cv2
import numpy as np
import matplotlib.pyplot as plt

def histogram_matching(source_image, target_image):
    source_hist, _ = np.histogram(source_image.flatten(), 256, [0,256])
    target_hist, _ = np.histogram(target_image.flatten(), 256, [0,256])

    # 计算累积直方图
    source_cdf = source_hist.cumsum()
    target_cdf = target_hist.cumsum()
    source_cdf_normalized = (source_cdf - source_cdf.min()) / (source_cdf.max() - source_cdf.min())
    target_cdf_normalized = (target_cdf - target_cdf.min()) / (target_cdf.max() - target_cdf.min())

    # 创建一个查找表，将源像素值映射到目标像素值
    lookup_table = np.interp(source_cdf_normalized, target_cdf_normalized, range(256))

    # 应用查找表进行直方图匹配
    matched_image = lookup_table[source_image]

    return matched_image

# 读取源图像和目标图像
source_image = cv2.imread('source_image.jpg', 0)
target_image = cv2.imread('target_image.jpg', 0)

# 进行直方图匹配
matched_image = histogram_matching(source_image, target_image)

# 显示结果
plt.figure(figsize=(10, 5))
plt.subplot(131), plt.imshow(source_image, cmap='gray'), plt.title('Source Image')
plt.subplot(132), plt.imshow(target_image, cmap='gray'), plt.title('Target Image')
plt.subplot(133), plt.imshow(matched_image, cmap='gray'), plt.title('Matched Image')
plt.show()

在这个示例中，`source_image`是原始图像，`target_image`是目标图像，函数`histogram_matching`实现了直方图匹配的算法。最后通过Matplotlib库将原始图像、目标图像和匹配后的图像进行显示。

请注意，这只是一个简单的示例代码，实际应用中可能需要进行更多的图像预处理和优化。