直方图匹配的python 代码

直方图匹配是一种图像处理技术，通过改变源图像的颜色分布来使其更接近目标图像的分布。在Python中，我们可以使用`skimage`库来实现这个过程。下面是一个简单的例子：

from skimage import exposure
import numpy as np

# 假设src_image和ref_image分别是源图像和参考图像
src_image = ... # 你的源图像数据
ref_image = ... # 你的参考图像数据

# 获取两个图像的直方图
src_hist, _ = np.histogram(src_image.flatten(), bins='auto')
ref_hist, _ = np.histogram(ref_image.flatten(), bins='auto')

# 使用归一化的方法找到最大值相同的索引
match_indices = exposure.match_histograms(src_hist, ref_hist, method='histogram')

# 应用匹配后的直方图到源图像
matched_image = exposure.rescale_intensity(src_image, in_range=(np.min(match_indices), np.max(match_indices)))

matched_image = matched_image.astype(np.uint8)  # 如果需要将结果转换回uint8类型

print("直方图匹配后的图像:")
display(matched_image)