灰度直方图生成算法*

算法原理

• 以CV_LOAD_IMAGE_GRAYSCALE模式读入图片
• 将图片从2维展平为1维向量
• 扫描1维向量，统计各灰度的像素个数（也可以直接遍历2维图片进行统计）
• plt.plot绘制图像

算法分析

时间复杂度 O( w*h )
w,h分别是图像的宽、长

例程（Python）

#-*-coding:utf-8-*-
import numpy as np
import cv2
import matplotlib.pyplot as plt

def calcGrayHist(img):
    rows,cols = img.shape
    img = img.reshape([rows*cols,])#展平
    grayHist = np.zeros([256],np.uint64)#灰度统计向量
    for pixel in range(img.shape[0]):#统计
        grayHist[img[pixel]] += 1
    return grayHist

if __name__ == '__main__':
    img = cv2.imread('timg.jpg',0)

    grayHist = calcGrayHist(img)

    plt.plot(grayHist)
    plt.xlabel(u'gray level')
    plt.ylabel(u'number of pixels')
    y_max = np.max(grayHist)
    plt.axis([0, 255, 0, y_max])
    plt.show()

    cv2.waitKey(0)
    cv2.destroyAllWindows()