Python灰度图像均衡化

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

#读取同文件夹下的lena图像并做灰度化处理
lena = plt.imread('lena.jpg')
lena = lena[:,:,1]
#显示原图（灰度图像）及其灰度直方图
plt.figure(1)
plt.subplot(221)
plt.imshow(lena,cmap='gray')
plt.subplot(223)
le = lena.flatten()
imhist = plt.hist(le, bins=256, facecolor='black', alpha=0.75)

equ = cv2.equalizeHist(lena)  # 得到直方图均衡化后的图像

# 绘制直方图均衡化后的图像及直方图
plt.figure(1)
plt.subplot(222)
plt.imshow(equ,cmap='gray')
plt.subplot(224)
eq = equ.flatten()
imhist2 = plt.hist(eq, bins=256, facecolor='black', alpha=0.75)
#显示图像
plt.show()