Python绘制图像的灰度直方图累计直方图

Posted 2022-02-05 安岳第二帅

        在计算机视觉中，对图像进行预处理时经常会用到图像的直方图、累计直方图等。下面给出一种参考代码，用python绘制图像灰度直方图和累计直方图。

       

# -*- coding: utf-8 -*-
"""
Created on Mon Jan 10 13:37:32 2022
@author: 2540817538（有问题请联系该QQ）
python3.8.8
"""

import numpy as np
import matplotlib.pyplot as plt
from PIL import Image

def histogram(grayfig):#绘制直方图
    x = grayfig.size[0]
    y = grayfig.size[1]
    ret = np.zeros(256)
    for i in range(x):  #遍历像素点获得灰度值
        for j in range(y):
            k = grayfig.getpixel((i,j))
            ret[k] = ret[k]+1
    for k in range(256):
        ret[k] = ret[k]/(x*y)
    return ret#返回包含各灰度值占比的数组

def histogram_sum(grayfig):#绘制累计直方图
    x = grayfig.size[0]
    y = grayfig.size[1]
    ret = np.zeros(256)
    for i in range(x):
        for j in range(y):
            k = grayfig.getpixel((i,j))
            ret[k] = ret[k]+1
    for k in range(1,256):
        ret[k] = ret[k]+ret[k-1]#累加
    for k in range(256):
        ret[k] = ret[k]/(x*y)
    return ret

im = Image.open('C:/Users/25408/Desktop/p1.jpg')#注意更改路径
im.show()
im_gray = im.convert('L')#获得灰度图
im_gray.show()

lenaGrayHist_1 = histogram(im_gray)
lenaGrayHist_2 = histogram_sum(im_gray)

plt.figure()
plt.rcParams['font.sans-serif'] = ['SimHei']#用来正常显示中文标签
plt.title("普通直方图")
plt.bar(range(256),lenaGrayHist_1,color='y')

plt.figure()
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.title("累计直方图")
plt.bar(range(256),lenaGrayHist_2)

效果如下：

输入的图像：

绘制结果：