我们要对灰度图像处理,首先定义直方图均衡化处理函数。
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图像才可以。