基于深度學(xué)習(xí)YOLO系列算法的圖像檢測

目前，基于深度學(xué)習(xí)算法的一系列目標(biāo)檢測算法大致可以分為兩大流派：

兩步走（two-stage）算法：先產(chǎn)生候選區(qū)域然后再進(jìn)行CNN分類(RCNN系列)

一步走（one-stage）算法：直接對(duì)輸入圖像應(yīng)用算法并輸出類別和相應(yīng)的定位(YOLO系列)

1YOLO算法的提出

在圖像的識(shí)別與定位中，輸入一張圖片，要求輸出其中所包含的對(duì)象，以及每個(gè)對(duì)象的位置（包含該對(duì)象的矩形框）。

對(duì)象的識(shí)別和定位，可以看成兩個(gè)任務(wù)：找到圖片中某個(gè)存在對(duì)象的區(qū)域，然后識(shí)別出該區(qū)域中具體是哪個(gè)對(duì)象。對(duì)象識(shí)別這件事（一張圖片僅包含一個(gè)對(duì)象，且基本占據(jù)圖片的整個(gè)范圍），最近幾年基于CNN卷積神經(jīng)網(wǎng)絡(luò)的各種方法已經(jīng)能達(dá)到不錯(cuò)的效果了。所以主要需要解決的問題是，對(duì)象在哪里。最簡單的想法，就是遍歷圖片中所有可能的位置，地毯式搜索不同大小，不同寬高比，不同位置的每個(gè)區(qū)域，逐一檢測其中是否存在某個(gè)對(duì)象，挑選其中概率最大的結(jié)果作為輸出。

顯然這種方法效率太低。RCNN開創(chuàng)性的提出了候選區(qū)(Region Proposals)的方法，先從圖片中搜索出一些可能存在對(duì)象的候選區(qū)（Selective Search），大概2000個(gè)左右，然后對(duì)每個(gè)候選區(qū)進(jìn)行對(duì)象識(shí)別，總體來說，RCNN系列依然是兩階段處理模式：先提出候選區(qū)，再識(shí)別候選區(qū)中的對(duì)象，大幅提升了對(duì)象識(shí)別和定位的效率。不過RCNN的速度依然很慢，其處理一張圖片大概需要49秒。因此又有了后續(xù)的Fast RCNN 和 Faster RCNN，針對(duì) RCNN的神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)和候選區(qū)的算法不斷改進(jìn)，F(xiàn)aster RCNN已經(jīng)可以達(dá)到一張圖片約0.2秒的處理速度。

R-CNN系列雖然準(zhǔn)確率比較高，但是即使是發(fā)展到Faster R-CNN，檢測一張圖片如下圖所示也要7fps(原文為5fps)，為了使得檢測的工作能夠用到實(shí)時(shí)的場景中，提出了YOLO。

2 算法的簡介

YOLO意思是You Only Look Once，創(chuàng)造性的將候選區(qū)和對(duì)象識(shí)別這兩個(gè)階段合二為一，看一眼圖片（不用看兩眼哦）就能知道有哪些對(duì)象以及它們的位置。

實(shí)際上，YOLO并沒有真正去掉候選區(qū)，而是采用了預(yù)定義的候選區(qū)（準(zhǔn)確點(diǎn)說應(yīng)該是預(yù)測區(qū)，因?yàn)椴⒉皇荈aster RCNN所采用的Anchor）。也就是將圖片劃分為 7*7=49 個(gè)網(wǎng)格（grid），每個(gè)網(wǎng)格允許預(yù)測出2個(gè)邊框（bounding box，包含某個(gè)對(duì)象的矩形框），總共 49*2=98 個(gè)bounding box。可以理解為98個(gè)候選區(qū)，它們很粗略的覆蓋了圖片的整個(gè)區(qū)域。

RCNN雖然會(huì)找到一些候選區(qū)，但畢竟只是候選，等真正識(shí)別出其中的對(duì)象以后，還要對(duì)候選區(qū)進(jìn)行微調(diào)，使之更接近真實(shí)的bounding box。這個(gè)過程就是邊框回歸：將候選區(qū)bounding box調(diào)整到更接近真實(shí)的bounding box。既然反正最后都是要調(diào)整的，干嘛還要先費(fèi)勁去尋找候選區(qū)呢，大致有個(gè)區(qū)域范圍就行了，所以YOLO就這么干了。

邊框回歸為什么能起作用，本質(zhì)上是因?yàn)榉诸?a target="_blank">信息中已經(jīng)包含了位置信息。就像你看到一只貓的臉和身體，就能推測出耳朵和屁股的位置。

3 算法的結(jié)構(gòu)

去掉候選區(qū)這個(gè)步驟以后，YOLO的結(jié)構(gòu)非常簡單，就是單純的卷積、池化最后加了兩層全連接。單看網(wǎng)絡(luò)結(jié)構(gòu)的話，和普通的CNN對(duì)象分類網(wǎng)絡(luò)幾乎沒有本質(zhì)的區(qū)別，最大的差異是最后輸出層用線性函數(shù)做激活函數(shù)，因?yàn)樾枰A(yù)測bounding box的位置（數(shù)值型），而不僅僅是對(duì)象的概率。所以粗略來說，YOLO的整個(gè)結(jié)構(gòu)就是輸入圖片經(jīng)過神經(jīng)網(wǎng)絡(luò)的變換得到一個(gè)輸出的張量，如下圖所示。

4 輸入輸出映射關(guān)系

因?yàn)橹皇且恍┏Ｒ?guī)的神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu)，所以，理解YOLO的設(shè)計(jì)的時(shí)候，重要的是理解輸入和輸出的映射關(guān)系。

參考圖5，輸入就是原始圖像，唯一的要求是縮放到448*448的大小。主要是因?yàn)閅OLO的網(wǎng)絡(luò)中，卷積層最后接了兩個(gè)全連接層，全連接層是要求固定大小的向量作為輸入，所以倒推回去也就要求原始圖像有固定的尺寸。那么YOLO設(shè)計(jì)的尺寸就是448*448。輸出是一個(gè) 7*7*30 的張量（tensor）。根據(jù)YOLO的設(shè)計(jì)，輸入圖像被劃分為 7*7 的網(wǎng)格（grid），輸出張量中的 7*7 就對(duì)應(yīng)著輸入圖像的 7*7 網(wǎng)格?；蛘呶覀儼?7*7*30 的張量看作 7*7=49個(gè)30維的向量，也就是輸入圖像中的每個(gè)網(wǎng)格對(duì)應(yīng)輸出一個(gè)30維的向量。參考上面圖5，比如輸入圖像左上角的網(wǎng)格對(duì)應(yīng)到輸出張量中左上角的向量。

要注意的是，并不是說僅僅網(wǎng)格內(nèi)的信息被映射到一個(gè)30維向量。經(jīng)過神經(jīng)網(wǎng)絡(luò)對(duì)輸入圖像信息的提取和變換，網(wǎng)格周邊的信息也會(huì)被識(shí)別和整理，最后編碼到那個(gè)30維向量中。

具體來看每個(gè)網(wǎng)格對(duì)應(yīng)的30維向量中包含了哪些信息。

① 20個(gè)對(duì)象分類的概率

因?yàn)閅OLO支持識(shí)別20種不同的對(duì)象（人、鳥、貓、汽車、椅子等），所以這里有20個(gè)值表示該網(wǎng)格位置存在任一種對(duì)象的概率。也對(duì)應(yīng)為20個(gè)object條件概率。

② 2個(gè)bounding box的位置

每個(gè)bounding box需要4個(gè)數(shù)值來表示其位置，(Center_x,Center_y,width,height)，即(bounding box的中心點(diǎn)的x坐標(biāo)，y坐標(biāo)，bounding box的寬度，高度)，2個(gè)bounding box共需要8個(gè)數(shù)值來表示其位置。

③ 2個(gè)bounding box的置信度

bounding box的置信度 = 該bounding box內(nèi)存在對(duì)象的概率 * 該bounding box與該對(duì)象實(shí)際bounding box的IOU, IOU=交集部分面積/并集部分面積，2個(gè)box完全重合時(shí)IOU=1，不相交時(shí)IOU=0。

綜合來說，一個(gè)bounding box的置信度Confidence意味著它 是否包含對(duì)象且位置準(zhǔn)確的程度。置信度高表示這里存在一個(gè)對(duì)象且位置比較準(zhǔn)確，置信度低表示可能沒有對(duì)象 或者 即便有對(duì)象也存在較大的位置偏差。作為監(jiān)督學(xué)習(xí)，我們需要先構(gòu)造好訓(xùn)練樣本，才能讓模型從中學(xué)習(xí)。對(duì)于一張輸入圖片，其對(duì)應(yīng)輸出的7*7*30張量（也就是通常監(jiān)督學(xué)習(xí)所說的標(biāo)簽y或者label）應(yīng)該填寫什么數(shù)據(jù)呢。

首先，輸出的 7*7維度 對(duì)應(yīng)于輸入的 7*7 網(wǎng)格。

① 20個(gè)對(duì)象分類的概率

② 2個(gè)bounding box的位置

③ 2個(gè)bounding box的置信度

（請(qǐng)對(duì)照上面圖6）

5 損失函數(shù)

損失就是網(wǎng)絡(luò)實(shí)際輸出值與樣本標(biāo)簽值之間的偏差。

YOLO給出的損失函數(shù)如下。

6 預(yù)測

訓(xùn)練好的YOLO網(wǎng)絡(luò)，輸入一張圖片，將輸出一個(gè) 7*7*30 的張量（tensor）來表示圖片中所有網(wǎng)格包含的對(duì)象（概率）以及該對(duì)象可能的2個(gè)位置（bounding box）和可信程度（置信度）。為了從中提取出最有可能的那些對(duì)象和位置，YOLO采用NMS（Non-maximal suppression，非極大值抑制）算法。

7 總結(jié)

YOLO以速度見長，處理速度可以達(dá)到45fps，其YoloV4版本（網(wǎng)絡(luò)較?。┥踔量梢赃_(dá)到155fps。這得益于其識(shí)別和定位合二為一的網(wǎng)絡(luò)設(shè)計(jì)，而且這種統(tǒng)一的設(shè)計(jì)也使得訓(xùn)練和預(yù)測可以端到端的進(jìn)行，非常簡便。不足之處是小對(duì)象檢測效果不太好（尤其是一些聚集在一起的小對(duì)象），對(duì)邊框的預(yù)測準(zhǔn)確度不是很高，總體預(yù)測精度略低于Fast RCNN。主要是因?yàn)榫W(wǎng)格設(shè)置比較稀疏，而且每個(gè)網(wǎng)格只預(yù)測兩個(gè)邊框，另外Pooling層會(huì)丟失一些細(xì)節(jié)信息，對(duì)定位存在影響。

責(zé)任編輯：xj

原文標(biāo)題：基于YOLO系列算法的圖像檢測

文章出處：【微信公眾號(hào)：新機(jī)器視覺】歡迎添加關(guān)注！文章轉(zhuǎn)載請(qǐng)注明出處。