直方图均衡化

我们要对灰度图像处理，首先定义直方图均衡化处理函数。

def histeq(img,nbr_bins=256):
    """ Histogram equalization of a grayscale image. """
    
    # 获取直方图p(r)
    imhist, bins = histogram(img.flatten(), nbr_bins, normed = True)
    
    # 获取T(r)
    cdf = imhist.cumsum() # cumulative distribution function
    cdf = 255 * cdf /cdf[-1] 
    
    # 获取s，并用s替换原始图像对应的灰度值
    result = interp(img.flatten(),bins[:-1],cdf)
    
    return result.reshape(img.shape),cdf

之后处理图像：

from pylab import *
from numpy import *
from PIL import Image

im = array(Image.open(r"/home/einstellung/桌面/钢丝绳图像分析/change.jpg").convert('L'))
im2, cdf = histeq(im)
imshow(im2)

保存图像：

Image.fromarray(im2).convert('RGB').save(r"/home/einstellung/桌面/钢丝绳图像分析/constrast_change.png")

因为这个是灰度图像，要想保存必须转换为RGB图像才可以。