python3 画灰度直方图

 

import numpy as np
import sys
import matplotlib.pyplot as plt
import cv2

def calcGrayHist(image):
    rows, cols = image.shape
    grayHist = np.zeros([256], np.uint64)
    for r in range(rows):
        for c in range(cols):
            grayHist[image[r][c]] += 1

    return grayHist


if __name__ == "__main__":
    #if len(sys.argv) > 1:
    #    image = cv2.imread(sys.argv[1], cv2.IMREAD_GRAYSCALE)
    #else:
    #    print("sys.argv null")
    image = cv2.imread("./2.png", cv2.IMREAD_GRAYSCALE)

    grayHist = calcGrayHist(image)

    x_range = range(256)
    plt.plot(x_range, grayHist, 'r', linewidth=2, c='red')
    y_maxValue = np.max(grayHist)
    plt.axis([0, 255, 0, y_maxValue])
    plt.xlabel("gray level")
    plt.ylabel("num of pixles")
    plt.show()