深入了解Python二維直方圖

更新時間：2021年12月25日 15:40:02 作者：機器視覺小學徒

只統(tǒng)計像素的灰度值這一特征，可將其成為一維直方圖。二維直方圖可以統(tǒng)計像素的色相和飽和度，用于查找圖像的顏色直方圖。本文將為大家介紹分別使用OpenCV和NumPy函數(shù)計算直方圖，需要的可以學習一下

前言

只統(tǒng)計像素的灰度值這一特征，可將其成為一維直方圖。二維直方圖可以統(tǒng)計像素的色相和飽和度，用于查找圖像的顏色直方圖。

一、OpenCV中的二維直方圖

OpenCV仍然使用cv2.calcHist()函數(shù)來查找圖像的顏色直方圖，只是在指定參數(shù)時與之前有所區(qū)別。

cv2.calcHist()函數(shù)的基本格式如下：

hist =cv2.calcHist(image, channels, mask, histSize, ranges)

image參數(shù)指定的原圖像應從BGR色彩空間轉換為HSV色彩空間, 實際參數(shù)需要用方括號括起來

channels參數(shù)設置為[0,1]時, 表示同時處理色相和飽和度

histSize參數(shù)設置BINS值為[180,256]時, 表示色相為180, 飽和度為256

ranges參數(shù)設置為[0,180,0,256]時, 表示色相的取值范圍為[0,180], 飽和度的取值范圍為[0,2565]

cv2.calcHist()函數(shù)返回的顏色直方圖可以直接使用cv2.show()函數(shù)顯示。

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

img = cv2.imread('XIAN.jpg')
cv2.namedWindow('orininal', cv2.WINDOW_NORMAL)
cv2.imshow('orininal', img)

img2 = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
hist = cv2.calcHist([img2], [0, 1], None, [180, 256], [0,180,0,256])

cv2.namedWindow('2DHist', cv2.WINDOW_NORMAL)
cv2.imshow('2DHist', hist)

cv2.waitKey(0)
cv2.destroyAllWindows()

plt.imshow(hist, interpolation = 'nearest')    # 繪制顏色直方圖
plt.show()                                     # 顯示顏色直方圖

cv2.calcHist()函數(shù)返回的顏色直方圖是一個大小為180*256的二維數(shù)組，用cv2.imshow()函數(shù)顯示時是一副灰度圖像，不能直接顯示出顏色的分布情況。

可以使用matplotlib.pyplot.imshow()函數(shù)繪制具有不同顏色的二維直方圖。

二、Numpy中的二維直方圖

Numpy的np.histogram2d()函數(shù)用于計算二維直方圖，其基本格式如下：

hist, xedges, yedges = np.histogram2d(x, y, bins, range)

hist為返回的直方圖

xedges為返回的x的直方圖的BINS邊界值

yedges為返回的y的直方圖的BINS邊界值

x和y為原圖對應通道轉換成的一維數(shù)組

bins為BINS的值, 如[180,256]

range為像素范圍, 格式為[[0, 180],[0, 256]]

img = cv2.imread('building.jpg')
cv2.imshow('orininal', img)

img2 = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
h, s, v = cv2.split(img2)

hist, x, y = np.histogram2d(h.ravel(), s.ravel(), [180, 256], [[0, 180],[0, 256]])

cv2.imshow('2DHist', hist)

cv2.waitKey(0)
cv2.destroyAllWindows()

plt.imshow(hist, interpolation = 'nearest')
plt.show()

可以使用matplotlib.pyplot.imshow()函數(shù)繪制具有不同顏色的二維直方圖。

三、直方圖示例

1、使用Numpy函數(shù)計算直方圖

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

img = cv2.imread('home.jpg')

plt.figure(figsize = (25,25))
imgrgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)        
plt.subplot(2, 2, 1)                                  
plt.title('Original')
plt.axis('off')
plt.imshow(imgrgb)

histb, e1 = np.histogram(img[0].ravel(), 256, [0, 255])   #計算B通道直方圖
histg, e2 = np.histogram(img[1].ravel(), 256, [0, 255])   #計算G通道直方圖
histr, e3 = np.histogram(img[2].ravel(), 256, [0, 255])   #計算R通道直方圖
plt.subplot(2, 2, 2)
plt.plot(histb, color = 'b')              
plt.plot(histg, color = 'g')              
plt.plot(histr, color = 'r')              
plt.title('Hist')


img2 = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)           #轉換色彩空間為HSV
h, s, v = cv2.split(img2)
hist, x, y=np.histogram2d(h.ravel(), s.ravel(), [180, 256], [[0, 180], [0, 256]])        #計算顏色直方圖
plt.subplot(2, 2, 3)
plt.title('2Dhist')                                 #設置子圖窗口標題
plt.imshow(hist, interpolation = 'nearest',cmap = 'gray')                                    #繪制顏色直方圖
plt.show()                                          #顯示顏色直方圖

2、使用OpenCV函數(shù)計算直方圖

# 2.使用OpenCV函數(shù)計算直方圖
import cv2
import numpy as np
import matplotlib.pyplot as plt

img = cv2.imread('flower.jpg')

plt.figure(figsize = (25,25))
imgrgb=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)         
plt.subplot(2, 2, 1)                           
plt.imshow(imgrgb)
plt.title('Original')                         
plt.axis('off')

histb = cv2.calcHist([img], [0], None, [256], [0,255])   #計算B通道直方圖
histg = cv2.calcHist([img], [1], None, [256], [0,255])   #計算G通道直方圖
histr = cv2.calcHist([img], [2], None, [256], [0,255])   #計算R通道直方圖
plt.subplot(2, 2, 2)
plt.plot(histb, color= 'b') 
plt.plot(histg, color= 'g') 
plt.plot(histr, color= 'r')  
plt.title('Hist') 


img2=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
hist = cv2.calcHist([img2], [0, 1], None, [180, 256], [0, 180, 0, 256])   
plt.subplot(2, 2, 3)
plt.title('2Dhist')                                 #設置子圖窗口標題
plt.imshow(hist,interpolation = 'nearest',cmap = 'gray')                                    #繪制顏色直方圖
plt.show()                                          #顯示顏色直方圖