图像直方图原理与Python实现

图像灰度直方图用来统计0到255的各灰度值在图像中的出现频次，也就是有多少个像素的值为0，多少个像素的值为1，等等。对于彩色图像，直方图为红、绿、蓝各分量的统计结果。

from os.path import isfile

from PIL import Image

def his(imageFile):

    #确保参数为图像文件

    assert isfile(imageFile) and imageFile.endswith(('.bmp', '.jpg', '.png')), 'Must be image file'

    #打开图像文件

    im = Image.open(imageFile)

    #获取图像颜色深度

    c = im.getpixel((0,0))

    #默认为彩色图像

    flag = False

    #灰度图像

    if isinstance(c, int):

        flag = True

    r = [0] * 256

    g = [0] * 256

    b = [0] * 256

    #获取图像尺寸

    width, height = im.size

    for w in range(width):

        for h in range(height):

            #读取像素颜色，并统计各分量值的频次

            c = im.getpixel((w,h))

            if flag:

                r[c] += 1

            else:

                r[c[0]] += 1

                g[c[1]] += 1

                b[c[2]] += 1

    if flag:

        return r

    else:

        return r+g+b

def his1(imageFile):

    #真正使用时建议直接直接使用pillow库中Image对象的histogram()获取直方图

    im = Image.open(imageFile)

    return im.histogram()

print(his('6.jpg')==his1('6.jpg'))