直方图均衡python代码

直方图均衡是一种常用的图像处理方法,用于改善图像的对比度和亮度分布。它以直方图作为基础,通过重新分配图像的灰度级,使图像中的像素值均匀分布在整个灰度级范围内,从而增强图像的视觉效果。

在图像处理中,直方图是一种统计图。它可以展示图像中每个灰度级的像素数量。直方图均衡的目标是使图像的像素分布在整个灰度级范围内尽可能均匀,使整个图像的对比度得到增强。

直方图均衡的原理是将图像的累积分布函数(CDF)映射到一个线性均匀分布的函数,进而改变图像的像素值。具体步骤如下:

1. 计算图像的直方图。通过遍历图像的每个像素,统计每个灰度级出现的次数,得到直方图。

2. 计算直方图的累积分布函数(CDF)。CDF表示从灰度级0到当前灰度级的像素数量累积和。

3. 使用CDF对原始图像进行均衡化。对于每个像素,将其灰度级映射到[0,255]范围内。映射规则为:新灰度级 = round((CDF[当前灰度级] - CDF[最小灰度级]) / ((M*N) - CDF[最小灰度级]) * (L-1)),其中M、N为图像的宽度和高度,L为灰度级范围。

4. 重新计算均衡后的图像的直方图。确认直方图均衡是否达到预期效果,即像素在图像中均匀分布。

5. 将均衡后的图像进行显示或保存。

下面是一个基于Python的直方图均衡的示例代码:

```python

import cv2

import numpy as np

def histogram_equalization(image):

# 计算直方图

hist, bins = np.histogram(image.flatten(), 256, [0,256])

# 计算累积分布函数

cdf = hist.cumsum()

cdf_normalized = cdf * hist.max() / cdf.max()

# 对原始图像进行均衡化

equalized_image = cdf_normalized[image]

# 重新计算均衡后图像的直方图

hist_equalized, _ = np.histogram(equalized_image.flatten(), 256, [0,256])

return equalized_image, hist, hist_equalized

# 读取图像

image = cv2.imread('image.jpg', 0) # 灰度图像

# 进行直方图均衡

equalized_image, hist, hist_equalized = histogram_equalization(image)

# 显示图像及其直方图

cv2.imshow('Original Image', image)

cv2.imshow('Equalized Image', equalized_image)

plt.plot(hist, color='r', label='Original Histogram')

plt.plot(hist_equalized, color='b', label='Equalized Histogram')

plt.legend()

plt.show()

# 保存均衡后的图像

cv2.imwrite('equalized_image.jpg', equalized_image)

```

上述代码首先通过 `np.histogram` 计算了原始图像的直方图,使用 `hist.cumsum()` 计算了累积分布函数(CDF)。

接着将原始图像的每个像素值通过 CDF 进行均衡化,得到了均衡后的图像 `equalized_image`。

最后,分别使用 `np.histogram` 计算了均衡后图像的直方图,并使用 `cv2.imshow` 显示了原始图像和均衡后图像,并使用 `plt.plot` 绘制了它们的直方图。

需要注意的是,直方图均衡化可能会导致一些细节的损失,特别是在图像局部对比度较高的情况下。因此,在应用直方图均衡化时,需要谨慎评估图像的特征和需求,选择适合的图像增强方法。

在实际应用中,直方图均衡化经常被用于图像增强、图像预处理、图像分割等任务。它不仅可以用于灰度图像,也可以用于彩色图像的亮度分量上。同时,它也是其他一些图像处理算法的基础,比如直方图匹配。

总结起来,直方图均衡是一种常用的图像处理方法,通过重新分配图像的灰度级来增强图像的对比度和亮度分布。它可以应用于各种图像处理任务中,但需要注意对图像细节的损失,并根据实际需求选择适合的图像增强方法。以上是一个基于Python的直方图均衡的简单示例代码,可以进一步优化和应用到实际项目中。 如果你喜欢我们三七知识分享网站的文章, 欢迎您分享或收藏知识分享网站文章 欢迎您到我们的网站逛逛喔!https://www.ynyuzhu.com/

点赞(17) 打赏

评论列表 共有 0 条评论

暂无评论
立即
投稿
发表
评论
返回
顶部