腳本之家服務器常用軟件

快捷導航

python?OpenCV圖像直方圖處理

更新時間：2022年06月27日 16:51:31 作者：吃貓的魚python

這篇文章主要介紹了python?OpenCV圖像直方圖處理，文章通過matplotlib畫一個直方圖展開詳情，具有一定的參考價值，需要的小伙伴可以參考一下

1.圖像直方圖基本含義和繪制

首先我們先要了解一下python三大劍客之一——matplotlib
我們都知道m(xù)atlab作為一個工具是公認的繪圖牛，但是我想說的是python下的matplotlib這個超級劍客也是非常厲害的，因為python近年來才火熱起來，所以熱度沒有matlib高，但是matlib可以實現(xiàn)的功能作為python都是差不多可以實現(xiàn)的。
我們回歸正題，先來介紹下matplotlib怎么簡單的畫一個直方圖。

import numpy as np
import matplotlib.pyplot as plt
x=np.arange(0,5,0.1)
y=np.sim(x)
plt.plot(x,y)

簡單的不能再簡單了吧，我們使用它去繪制一個sin（x）的函數(shù)圖像。

那么matplotlib如何和CV一起工作呢？

import cv2
import matplotlib.pyplot as plt
o=cv2.imread("image\\boatGray.bmp")
histb = cv2.calcHist([o],[0],None,[256],[0,255])
plt.plot(histb,color='b')
plt.show()

結果是（也很好理解吧）：

針對于彩色圖像我們也可以針對BGR分別作圖。

import cv2
import matplotlib.pyplot as plt
o=cv2.imread("image\\girl.bmp")
histb = cv2.calcHist([o],[0],None,[256],[0,255])
histg = cv2.calcHist([o],[1],None,[256],[0,255])
histr = cv2.calcHist([o],[2],None,[256],[0,255])
plt.plot(histb,color='b')
plt.plot(histg,color='g')
plt.plot(histr,color='r')
plt.show()

直方圖的繪制我們使用的函數(shù)還有：函數(shù)hist
功能：根據(jù)數(shù)據(jù)源和像素級繪制直方圖。
語法： hist(數(shù)據(jù)源，像素級）
數(shù)據(jù)源：圖像，必須是一維數(shù)組。
像素級：一般是256，指[0,255]

功能：將多維數(shù)組降為一維數(shù)組。格式：一維數(shù)組=多維數(shù)組.ravel()

import cv2
import matplotlib.pyplot as plt
o=cv2.imread("image\\boat.jpg")
cv2.imshow("o",o)
cv2.waitKey()
cv2.destroyAllWindows()
plt.hist(o.ravel(),256)

2.OpenCV統(tǒng)計直方圖并繪制

使用OpenCV對圖像進行繪制的和橫坐標表示像素值比如[0,255]，縱坐標表示像素值的個數(shù)。

繪制函數(shù)：

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

其中，hist返回結果是一個直方圖，返回的直方圖，是一個二維數(shù)組。
image原始圖像：圖像需要使用“[ ]”括起來使用。
channels：

通道編號需要用中括號括起來輸入圖像是灰度圖時，它的值是[0]；彩色圖像可以是[0]，[1]，[2]。分別對應通道B，G，R。
mask：掩碼圖像統(tǒng)計整幅圖像的直方圖，設為None。統(tǒng)計圖像某一部分的直方圖時，需要掩碼圖像。
histSize
BINS的數(shù)量，例如【256】
ranges
像素值范圍RANGE
accumulate默認值為false。如果被設置為true，則直方圖在開始分配時不會被清零。該參數(shù)允許從多個對象中計算單個直方圖，或者用于實時更新直方圖。多個直方圖的累積結果，用于對一組圖像計算直方圖。

使用OpenCV畫出直方圖：

import cv2
import matplotlib.pyplot as plt
o=cv2.imread("image\\girl.bmp")
histb = cv2.calcHist([o],[0],None,[256],[0,255])
histg = cv2.calcHist([o],[1],None,[256],[0,255])
histr = cv2.calcHist([o],[2],None,[256],[0,255])
plt.plot(histb,color='b')
plt.plot(histg,color='g')
plt.plot(histr,color='r')
plt.show()

3.使用掩碼的直方圖-直方圖、掩膜

掩碼說實話就是使用掩膜的黑色部分把原始圖像的部分給覆蓋掉，也稱為過濾掉。那么我們怎么做呢？首先我們需要創(chuàng)建一個掩膜：

mask=np.zeros(image.shape,np.uint8)
mask[200:400,200:400]=255

首先創(chuàng)建一個全0的和原圖像size一致的，然后我們把指定范圍指定為白色。然后傳入函數(shù)內：

import cv2
import numpy as np
import matplotlib.pyplot as plt
顯示直方圖
image=cv2.imread("image\\girl.bmp",cv2.IMREAD_GRAYSCALE)
mask=np.zeros(image.shape,np.uint8)
mask[200:400,200:400]=255
histMI=cv2.calcHist([image],[0],mask,[256],[0,255])
histImage=cv2.calcHist([image],[0],None,[256],[0,255])
plt.plot(histImage)
plt.plot(histMI)

結果是：

掩膜原理：

說實在的就是與和或的關系，與就是一個不行就都不行。或就是一個行就可以。

而我們的掩膜原理主要用到的就是與操作；

計算結果 = cv2.bitwise_and(圖像1，圖像2)

import cv2
import numpy as np
import matplotlib.pyplot as plt
image=cv2.imread("image\\boat.bmp",0)
mask=np.zeros(image.shape,np.uint8)
mask[200:400,200:400]=255
mi=cv2.bitwise_and(image,mask)
cv2.imshow('original',image)
cv2.imshow('mask',mask)
cv2.imshow('mi',mi)
cv2.waitKey()
cv2.destroyAllWindows()

4.直方圖均衡化原理及函數(shù)

在維基百科是這樣定義的：

對應在圖像上就是：

前提：如果一幅圖像占有全部可能的灰度級，并且均勻分布。
結論：該圖像具有高對比度和多變的灰度色調。
外觀：圖像細節(jié)豐富，質量更高。

算法：

1.計算累計直方圖
2.將累計直方圖進行區(qū)間轉換
3.在累計直方圖中，概率相近的原始值，會被處理為相同的值。

1.計算灰度級出現(xiàn)的概率情況

????：第K個灰度級
????：第k級灰度的像素個數(shù)
N：圖像內總像素的個數(shù)
L：灰度級最大值，灰度值區(qū)間[0,L-1]
2.變換函數(shù)

我們把公式表現(xiàn)在圖片上就是：

這樣就完成了從原始圖像計算得到了均衡直方圖。

雖然二者相似。但右側的分布更均衡，相鄰像素級概率和與高概率近似相等。可應用到醫(yī)療圖像處理，車牌識別，人臉識別。

對應函數(shù)是：dst = cv2.equalizeHist( src )

import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread('image\\equ.bmp',cv2.IMREAD_GRAYSCALE)
equ = cv2.equalizeHist(img)
plt.hist(img.ravel(),256)
plt.figure()
plt.hist(equ.ravel(),256)

我們處理一下lena，就是這樣：

5.子圖的繪制

我們有的時候為了方便對比，會想要把幾個圖放在一張大圖中進行比較，那么我們怎么去做呢？有沒有一個函數(shù)可以完成這個操作呢。
subplot(nrows, ncols, plot_number)
nrows表示行數(shù)，ncols表示列數(shù)，plot_number表示第幾個。subplot(2,3,4)那么就表示2行三列，第四個圖。
當每一個參數(shù)都小于10時，可以直接書寫三個數(shù)字，表示為“subplot(234)

import cv2
import numpy as np
import matplotlib.pyplot as plt
img = cv2.imread('image\\boatGray.bmp',cv2.IMREAD_GRAYSCALE)
equ = cv2.equalizeHist(img)
plt.subplot(121),plt.hist(img.ravel(),256)
plt.subplot(122),plt.hist(equ.ravel(),256)

imshow(X, cmap=None)
X表示要繪制的圖像，cmap表示colormap，顏色圖譜，默認為RGB(A)顏色空間。
灰度圖像：colormap，顏色圖譜，默認為RGB(A)顏色空間使用參數(shù)cmap=plt.cm.gray
彩色圖像：colormap，顏色圖譜，默認為RGB(A)顏色空間，如果使用opencv讀入的圖像，默認空間為BGR，需要調整色彩空間為RGB。

import cv2
import matplotlib.pyplot as plt
o = cv2.imread('image\\girl.bmp')
g=cv2.cvtColor(o, cv2.COLOR_BGR2GRAY)
plt.subplot(221)
plt.imshow(o),plt.axis('off')
plt.subplot(222)
plt.imshow(o,cmap=plt.cm.gray),plt.axis('off')
plt.subplot(223)
plt.imshow(g),plt.axis('off')
plt.subplot(224)
plt.imshow(g,cmap=plt.cm.gray),plt.axis('off')

第一個圖是：彩色圖像，使用默認參數(shù)。
第二個圖是：彩色圖像，使用參數(shù)cmap=plt.cm.gray
第三個圖是：灰色圖像，使用默認參數(shù)
第四個圖是：灰色圖像，使用參數(shù)cmap=plt.cm.gray
那么只有第四個圖是正確的。

對于彩色圖像：

import cv2
import matplotlib.pyplot as plt
img = cv2.imread('image\\girl.bmp')
b,g,r=cv2.split(img)
img2=cv2.merge([r,g,b])
plt.subplot(121)
plt.imshow(img),plt.axis('off')
plt.subplot(122)
plt.imshow(img2),plt.axis('off')

必須要將BGR split然后merge成RGB才可以。

6.直方圖均衡化對比

import cv2
import matplotlib.pyplot as plt
img = cv2.imread('image\\boat.bmp',cv2.IMREAD_GRAYSCALE)
equ = cv2.equalizeHist(img)
plt.subplot(221)
plt.imshow(img,cmap=plt.cm.gray),plt.axis('off')
plt.subplot(222)
plt.imshow(equ,cmap=plt.cm.gray),plt.axis('off')
plt.subplot(223)
plt.hist(img.ravel(),256)
plt.subplot(224)
plt.hist(equ.ravel(),256)