什么是方向梯度直方圖
在前面的實(shí)驗(yàn) 1髓考、實(shí)驗(yàn) 2 中部念,我們了解到傳統(tǒng)的目標(biāo)檢測流程可分為三個(gè)步驟,第一步是使用滑動窗口和圖像金字塔從圖片中選擇一些區(qū)域氨菇。第二步是將選擇出來的區(qū)域轉(zhuǎn)化為人工設(shè)計(jì)的特征儡炼,可稱為特征提取。第三步是將這些特征輸入分類器進(jìn)行分類查蓉。方向梯度直方圖(Histogram of Oriented Gradients)以下簡稱 HOG射赛,就是一種人工設(shè)計(jì)的特征,用來簡化圖像表述的特征描述符奶是。
下圖中左邊的圖片是一只貓楣责,我們不僅可以看出貓身體上的特征、顏色聂沙、紋理秆麸,而且還能看到背景。右邊的圖片是使用 HOG 來表示的圖片及汉,除了可以看到圖中能看出貓的外形沮趣,其他的細(xì)節(jié)包括背景幾乎都被去除了,故右邊的圖片是左邊圖片的一種簡化表示形式坷随。HOG 可以用來表示物體的形狀房铭、外形特征,將這些特征輸入分類器就可以實(shí)現(xiàn)目標(biāo)的分類温眉。
在傳統(tǒng)的算法中缸匪,使用 HOG 描述圖片,可以保留有用信息类溢,剔除無用的信息凌蔬,這樣不僅減少計(jì)算量,還使得分類器的效果更好闯冷。HOG 可以分為以下幾個(gè)步驟:預(yù)先歸一化砂心、計(jì)算水平和垂直方向梯度、計(jì)算梯度直方圖蛇耀、區(qū)域(Blocks)歸一化辩诞。
方向梯度直方圖流程
預(yù)先歸一化(Normalization)
在計(jì)算梯度前可對圖片歸一化(Normalization)處理,歸一化的目的是使所有的數(shù)值落入到統(tǒng)一的范圍內(nèi)纺涤,從而使算法能有更好的表現(xiàn)译暂。在 HOG 的原論文中提到使用伽馬矯正的方法處理輸入圖片抠忘,伽馬矯正可以增加圖像的對比度。但是在很多情況下秧秉,伽馬矯正對提升算法效果不明顯褐桌,我們可以跳過圖片預(yù)先歸一化衰抑,直接計(jì)算圖片梯度象迎。
計(jì)算梯度
前面提到可以不用預(yù)先對圖片進(jìn)行歸一化,故可以說 HOG 特征是從計(jì)算圖像在水平方向和垂直方向上的梯度開始的呛踊。圖像的梯度計(jì)算是使用卷積核對圖像進(jìn)行卷積操作砾淌,例如我們可以使用矩陣 [[-1, 0, 1]] 和 [[-1], [0], [1]] 分別與圖像上的每個(gè)像素進(jìn)行運(yùn)算來獲得水平和垂直方向上的梯度。
上面的兩個(gè)公式分別計(jì)算每個(gè)像素的梯度幅值G和梯度方向θ谭网。下面左圖是合并水平汪厨、垂直方向上的梯度獲得的梯度幅值,可以看到相較于水平愉择、垂直方向上的圖片劫乱,左圖中貓的輪廓更清晰明顯。右圖表示圖片中的梯度方向锥涕。
方向梯度直方圖
現(xiàn)在我們已經(jīng)有了梯度幅值G和梯度方向θ衷戈,接下來我們就可以計(jì)算方向梯度直方圖了。在計(jì)算方向梯度直方圖之前层坠,我們需要將圖片分成若干個(gè)小方格(Cells)殖妇,為避免歧義下文皆書寫為 Cell 或 Cells 。例如破花,下圖是一張寬高為649×385 的圖片谦趣,我們將其平均分割成若干個(gè) Cells,每個(gè) Cell 內(nèi)包含 8×8 個(gè)像素座每,所以圖片的高被分為 ?385÷8?=48 份前鹅,圖片的寬被分為 ?649÷8?=81 份(?? 表示向下取整),故整張圖片有48×81 個(gè) Cells峭梳。
至此我們已經(jīng)將圖片分成許多 Cells嫡纠,對于每一個(gè) Cell,使用G和θ 來構(gòu)建方向梯度直方圖延赌。首先我們先選擇梯度方向的范圍除盏,梯度方向的范圍可分為 0 到 180 度(無符號)和 0 到 360 度(有符號),通常使用 0 到 180 度的范圍挫以。然后將 0 到 180 度的范圍分成 9 個(gè)區(qū)間(bins)者蠕,分別為 0 到 20 度,20 到 40 度掐松, 40 到 60 度 …… 160 到 180 度踱侣。每個(gè)像素都有一個(gè)梯度幅值和一個(gè)梯度方向粪小,所以方向梯度直方圖的計(jì)算就是每個(gè)像素所對應(yīng)的梯度方向落在 9 個(gè)區(qū)間中的哪一個(gè),那么該像素的梯度幅值就在該區(qū)間中累加抡句。
下圖是一個(gè)計(jì)算方向梯度圖的例子探膊,對于紅色方框中的像素,假設(shè)其中有些像素對應(yīng)的梯度方向落在 0 到 20 區(qū)間待榔,那么將這些像素對應(yīng)的梯度幅值在 0 到 20 區(qū)間內(nèi)進(jìn)行累加逞壁,同理其他區(qū)間也做同樣的運(yùn)算,最終得到下圖中右邊的方向梯度直方圖锐锣。同樣地腌闯,整張圖片中的所有 Cells 都用同樣的方法計(jì)算方向梯度直方圖。
區(qū)域(Blocks)歸一化
我們已經(jīng)將圖像分成若干個(gè) Cells雕憔,并且計(jì)算了每個(gè) Cell 的方向梯度直方圖姿骏。接下來我們要對圖像進(jìn)行區(qū)域歸一化處理,歸一化的目的是減少光照變化對圖像梯度的影響斤彼。 現(xiàn)在讓我們來看看如何進(jìn)行歸一化處理分瘦。
首先我們先來介紹什么是區(qū)域(Blocks),為避免歧義下文皆書寫為 Block 或 Blocks琉苇,前面我們將圖像分成若干個(gè) Cells嘲玫,每個(gè) Cell 內(nèi)有若干個(gè)像素,類似地翁潘,一個(gè) Block 是一塊由若干個(gè) Cells 組成的矩形趁冈。對圖像進(jìn)行歸一化的過程類似于前面實(shí)驗(yàn)學(xué)習(xí)的滑動窗口,將一個(gè) Block 從左向右拜马、從上向下在圖中滑動渗勘,然后在每個(gè) Block 區(qū)域內(nèi)進(jìn)行歸一化計(jì)算。
讓我們通過下圖來理解如何通過 Blocks 對圖片進(jìn)行歸一化俩莽。 下圖中右邊圖片是我們從原圖片中選取的一部分旺坠,在這塊區(qū)域里面有若干個(gè) Cells,我們用紅色矩形框表示一個(gè) Blcok扮超,紅色矩形框在圖上向右滑動一個(gè) Cell 的步長后我們就得到了藍(lán)色矩形框取刃,所以區(qū)域歸一化的方法就是設(shè)定一個(gè)尺寸為K×K 個(gè) Cells 的 Block,在圖上從左向右出刷、從上向下滑動璧疗,然后在每個(gè) Block 內(nèi)進(jìn)行歸一化。
在每個(gè) Block 內(nèi)有2×2 個(gè) Cells馁龟,前面我們將梯度方向分為 9 個(gè)區(qū)間并為每個(gè) Cell 計(jì)算了方向梯度直方圖崩侠,故每個(gè) Cell 有 9 個(gè)向量财饥,則在一個(gè) Block 內(nèi)共有2×2×9 個(gè)向量根悼。然后我們可以使用L1范數(shù)或L2范數(shù)對 Block 內(nèi)的向量進(jìn)行歸一化抄腔。其中使用L2范數(shù)進(jìn)行歸一化的效果相對較好售淡,下面就是使用L2范數(shù)歸一化的公式,即 Block 內(nèi)的每個(gè)向量除以由 Block 內(nèi)所有向量計(jì)算得到的L2范數(shù)系瓢。其中vi表示 Block 內(nèi)的向量阿纤,? 的作用是防止出現(xiàn)分母為 0 的情況,它是一個(gè)很小的值夷陋。
從上圖中可以看出每一個(gè) Cell 不止出現(xiàn)在一個(gè) Block 內(nèi)欠拾,也就是說一個(gè) Cell 將被重復(fù)的用于歸一化計(jì)算中,這樣做會看似比較冗余肌稻,但是會提高特征描述的表現(xiàn)清蚀。最后對所有的 Block 完成歸一化計(jì)算匕荸,合并所有獲得的歸一化后的向量爹谭,這樣我們就完成了圖像的 HOG 特征化表示。
使用 Scikit-image 實(shí)現(xiàn)方向梯度直方圖
本節(jié)實(shí)驗(yàn)我們將通過幾行簡單的代碼來實(shí)現(xiàn) HOG 算法榛搔。首先我們執(zhí)行下面命令下載需要用到的圖片诺凡。
!wget https://labfile.oss.aliyuncs.com/courses/3096/pets.jpg
然后我們從 Scikit-image 導(dǎo)入 feature 和 exposure 模塊。feature 模塊里存放著一些用于計(jì)算特征的算法践惑,exposure 模塊具有一些直方圖處理的功能腹泌。我們還需要導(dǎo)入 cv2 模塊用于圖片的讀取。
from skimage import feature
from skimage import exposure
from matplotlib import pyplot as plt
import cv2
%matplotlib inline
接著使用 cv2.imread 函數(shù)讀取圖片尔觉。
image = cv2.imread("pets.jpg")
我們使用 feature.hog 用于計(jì)算圖片的方向梯度直方圖凉袱。該函數(shù)的參數(shù)意義如下所示。
第一個(gè)參數(shù) image 表示輸入圖像侦铜。
orientations 表示要將梯度方向分成幾個(gè)區(qū)間专甩,這里我們將梯度方向分為 9 個(gè)區(qū)間。
pixels_per_cell 表示 Cell 的尺寸钉稍,即一個(gè) Cell 中有幾個(gè)像素涤躲,需要傳遞一個(gè)元組給該參數(shù),我們將 (8, 8) 傳遞給該參數(shù)贡未。
cells_per_block 表示每個(gè) Block 的尺寸种樱,即一個(gè) Block 中有幾個(gè) Cells,這里需要傳遞一個(gè)元組給該參數(shù)俊卤,我們將 (2, 2) 傳遞給該參數(shù)嫩挤。
transform_sqrt 表示伽馬校正,我們將 True 傳遞給該參數(shù)表示使用伽馬校正預(yù)先對圖片進(jìn)行歸一化處理消恍。
visualize 表示可視化岂昭,將 True 傳遞給該參數(shù)表示返回 HOG 圖像。
(o, hog) = feature.hog(image, orientations=9, pixels_per_cell=(8, 8),
cells_per_block=(2, 2), transform_sqrt=True, visualize=True)
該函數(shù)返回 2 個(gè)值哺哼,第一個(gè)值 o 是 HOG 圖像的一維展開數(shù)組佩抹。第二個(gè)值 hog 表示返回一個(gè)高和寬同輸入圖像一樣的二維數(shù)組叼风,這個(gè)值可用于可視化方向梯度直方圖。
接下來我們使用 exposure.rescale_intensity 來調(diào)整的輸入圖片的像素值大小或像素強(qiáng)度棍苹。因?yàn)楂@得的 hog 中的元素值都為被歸一化了无宿,所以這些值都比較小,如果直接將這些值作為圖像的像素值枢里,那么該圖像看起來像一張全黑圖片孽鸡,我們需要用該函數(shù)將這些值拉伸到一個(gè)較大的范圍。
hog = exposure.rescale_intensity(hog, out_range=(0, 255))
hog = hog.astype("uint8")
hog[hog > 50] = 255
我們傳遞 2 個(gè)參數(shù)給該函數(shù)栏豺,第一個(gè)參數(shù) hog 表示前面獲取的可視化二維數(shù)組彬碱。第二個(gè)參數(shù) out_range 表示將輸入圖片的像素強(qiáng)度拉伸到設(shè)定的范圍,這里我們將 hog 中的每個(gè)元素值拉伸到 (0, 255) 范圍內(nèi)奥洼。接下來我們使用 astype 方法將 hog 中的元素轉(zhuǎn)換為 uint8 類型巷疼。最后我們將 hog 中大于 50 的元素賦值為 255。
我們使用 plt.figure 創(chuàng)建畫板灵奖,參數(shù) dpi=150 表示圖像的分辨率設(shè)置為 150嚼沿。最后使用 plt.imshow 函數(shù)顯示圖片。
plt.figure(dpi=150)
plt.imshow(hog, cmap = 'gray', aspect='auto')
如下圖瓷患,左邊圖片是輸入圖片骡尽,右邊是對輸入圖片使用 HOG 提取特征后的圖片,可以看到相較于左邊圖片擅编,右邊圖片僅保留了動物的外形特征攀细。