python實現圖像直方圖均衡

def histeq(img,nbr_bins=256):
    """ Histogram equalization of a grayscale image. """


    imhist, bins = np.histogram(img.flatten(), nbr_bins, normed = True)

    cdf = imhist.cumsum() # cumulative distribution function
    cdf = 255 * cdf /cdf[-1] 

    # 獲取s，並用s替換原始圖像對應的灰度值
    result = np.interp(img.flatten(),bins[:-1],cdf)

    return result.reshape(img.shape),cdf