Python如何实现均直方图均衡化

import cv2
import matplotlib.pyplot as plt
import numpy as np
# 定义matshow方法
def matshow(title='image', image=None, gray=False):
    if isinstance(image, np.ndarray):
        if len(image.shape) == 2:
            pass
        elif gray:
            # 转换成GRAY格式
            image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        else:
            # 图片默认BGR通道，将突破转换成RGB
            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
            # 使用这种方式显示图片可能会导致图片显示畸形
            # cv2.imshow('image', image)
    plt.figure()
    # 载入图像
    plt.imshow(image, cmap="gray")
    # 设置标题
    plt.title(title)
    plt.show()
if __name__ == '__main__':
    im = cv2.imread('data/lena.jpg', 0)
    matshow('im', im)
    # 绘制直方图
    plt.hist(im.ravel(), 256, [0, 256])
    plt.show()
    # 均衡化处理
    im_equ1 = cv2.equalizeHist(im)
    matshow('im_equ1', im_equ1)
    # 绘制均衡化处理的直方图
    plt.hist(im_equ1.ravel(), 256, [0, 256])
    plt.show()