直方图均衡化 python
时间: 2023-11-08 09:05:58 浏览: 37
直方图均衡化是一种常用的图像增强方法,可以提高图像的对比度和亮度。在Python中,可以使用OpenCV库来实现直方图均衡化。具体步骤如下:
1. 读取图像
2. 将图像转换为灰度图像
3. 计算灰度图像的直方图
4. 计算累积分布函数
5. 根据累积分布函数计算新的像素值
6. 将新的像素值赋给原始图像
7. 显示增强后的图像
下面是一个简单的示例代码:
```python
import cv2
# 读取图像
img = cv2.imread('image.jpg')
# 将图像转换为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 计算灰度图像的直方图
hist, bins = np.histogram(gray.flatten(), 256, [0, 256])
# 计算累积分布函数
cdf = hist.cumsum()
cdf_normalized = cdf * hist.max() / cdf.max()
# 根据累积分布函数计算新的像素值
new_gray = np.interp(gray.flatten(), bins[:-1], cdf_normalized)
# 将新的像素值赋给原始图像
new_gray = new_gray.reshape(gray.shape)
new_img = cv2.merge((new_gray, new_gray, new_gray))
# 显示增强后的图像
cv2.imshow('Enhanced Image', new_img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```