博客專欄

EEPW首頁 > 博客 > Anchor-free目標(biāo)檢測 | 工業(yè)應(yīng)用更友好的新網(wǎng)絡(luò)(3)

Anchor-free目標(biāo)檢測 | 工業(yè)應(yīng)用更友好的新網(wǎng)絡(luò)(3)

發(fā)布人:CV研究院 時(shí)間:2021-09-16 來源:工程師 發(fā)布文章

貢獻(xiàn):

1.通過檢測bbox的一對角點(diǎn)來檢測出目標(biāo)。

2.提出corner pooling,來更好的定位bbox的角點(diǎn)。

19.png

上圖是top-left corner的 Corner Pooling過程。在水平方向,從最右端開始往最左端遍歷,每個(gè)位置的值都變成從最右到當(dāng)前位置為止,出現(xiàn)的最大的值。同理,bottom-right corner的Corner Pooling則是最左端開始往最右端遍歷。同樣的,在垂直方向上,也是這樣同樣的Pooling的方式。

以左上角點(diǎn)為例,當(dāng)我們決定此點(diǎn)是否個(gè)corner點(diǎn)的時(shí)候,往往會沿著水平的方向向右看,看看是否與物體有相切,還會沿著垂直方向向下看,看看是否與物體相切。簡而言之,其實(shí)corner點(diǎn)是物體上邊緣點(diǎn)和坐邊緣點(diǎn)的集合,因此在pooling的時(shí)候通過Corner Pooling的方式能夠一定程度上體現(xiàn)出當(dāng)前點(diǎn)出發(fā)的射線是否與物體相交。

ExtremeNet:

20.jpg

作者使用了最佳的關(guān)鍵點(diǎn)估計(jì)框架,通過對每個(gè)目標(biāo)類預(yù)測4個(gè)多峰值的heatmaps來尋找極值點(diǎn)。另外,作者使用每個(gè)類center heatmap來預(yù)測目標(biāo)中心。僅通過基于幾何的方法來對極值點(diǎn)分組,如果4個(gè)極值點(diǎn)的幾何中點(diǎn)在center map上對應(yīng)的分?jǐn)?shù)高于閾值,則這4個(gè)極值點(diǎn)分為一組。

offset的預(yù)測是類別無關(guān)的,而極值點(diǎn)的預(yù)測是類別相關(guān)的。對每種極值點(diǎn)heatmap,不包含center map,預(yù)測2張offset map(分別對應(yīng)XY軸方向)。網(wǎng)絡(luò)的輸出是5xC heatmaps和4x2offset maps,C是類別數(shù)。

分組算法的輸入是每個(gè)類的5個(gè)heatmaps,一個(gè)center heatmap和4個(gè)extreme heatmaps,通過檢測所有的峰值來提取出5個(gè)heatmaps的關(guān)鍵點(diǎn)。給出4個(gè)極值點(diǎn),計(jì)算幾何中心,如果幾何中心在center map上對應(yīng)高響應(yīng),那么這4個(gè)極值點(diǎn)為有效檢測。作者使用暴力枚舉的方式來得到所有有效的4個(gè)關(guān)鍵點(diǎn)。

貢獻(xiàn):

1.將關(guān)鍵點(diǎn)定義為極值點(diǎn)。

2.根據(jù)幾何結(jié)構(gòu)對關(guān)鍵點(diǎn)進(jìn)行分組。

CornerNet和ExtremeNet的區(qū)別:

1.CornerNet通過預(yù)測角點(diǎn)來檢測目標(biāo)的,而ExtremeNet通過預(yù)測極值點(diǎn)和中心點(diǎn)來檢測目標(biāo)的。

2.CornerNet通過角點(diǎn)embedding之間的距離來判斷是否為同一組關(guān)鍵點(diǎn),而ExtremeNet通過暴力枚舉極值點(diǎn)、經(jīng)過中心點(diǎn)判斷4個(gè)極值點(diǎn)是否為一組。

FSAF:

讓每個(gè)實(shí)例選擇最好的特征層來優(yōu)化網(wǎng)絡(luò),因此不需要anchor來限制特征的選擇。

21.jpg

一個(gè)anchor-free的分支在每個(gè)特征金字塔層構(gòu)建,獨(dú)立于anchor-based的分支。和anchor-based分支相似,anchor-free分支由分類子網(wǎng)絡(luò)和回歸子網(wǎng)絡(luò)。一個(gè)實(shí)例能夠被安排到任意層的anchor-free分支。訓(xùn)練期間,基于實(shí)例的信息而不是實(shí)例box的尺寸來動(dòng)態(tài)地為每個(gè)實(shí)例選擇最合適的特征層。選擇的特征層學(xué)會檢測安排的實(shí)例。推理階段,F(xiàn)SAF模塊和anchor-based分支獨(dú)立或者聯(lián)合運(yùn)行。

22.jpg

在RetinaNet的基礎(chǔ)上,F(xiàn)SAF模塊引入了2個(gè)額外的卷積層,這兩個(gè)卷積層各自負(fù)責(zé)anchor-free分支的分類和回歸預(yù)測。具體的,在分類子網(wǎng)絡(luò)中,feature map后面跟著K個(gè)3x3的卷積層和sigmoid,在回歸子網(wǎng)絡(luò)中,feature map后面跟著4個(gè)3x3的卷積層和ReLU。

23.jpg

實(shí)例輸入到特征金字塔的所有層,然后求得所有anchor-free分支focal loss和IoU loss的和,選擇loss和最小的特征層來學(xué)習(xí)實(shí)例。訓(xùn)練時(shí),特征根據(jù)安排的實(shí)例進(jìn)行更新。推理時(shí),不需要進(jìn)行特征更新,因?yàn)樽詈线m的特征金字塔層自然地輸出高置信分?jǐn)?shù)。

FCOS:

24.png

和語義分割相同,檢測器直接將位置作為訓(xùn)練樣本而不是anchor。具體的,如果某個(gè)位置落入了任何gt中,那么該位置就被認(rèn)為是正樣本,并且類別為該gt的類別?;赼nchor的檢測器,根據(jù)不同尺寸安排anchor到不同的特征層,而FCOS直接限制邊界框回歸的范圍(即每個(gè)feature map負(fù)責(zé)一定尺度的回歸框)。

Center-ness:

25.png

為了剔除遠(yuǎn)離目標(biāo)中心的低質(zhì)量預(yù)測bbox,作者提出了添加center-ness分支,和分類分支并行。

26.png

優(yōu)點(diǎn):

1.將檢測和其他使用FCN的任務(wù)統(tǒng)一起來,容易重用這些任務(wù)的思想。

2.proposal free和anchor free,減少了超參的設(shè)計(jì)。

3.不使用trick,達(dá)到了單階段檢測的最佳性能。

4.經(jīng)過小的修改,可以立即拓展到其他視覺任務(wù)上。

FoveaBox:

人類眼睛的中央凹:視野(物體)的中心具有最高的視覺敏銳度。FoveaBox聯(lián)合預(yù)測對象中心區(qū)域可能存在的位置以及每個(gè)有效位置的邊界框。由于特征金字塔的特征表示,不同尺度的目標(biāo)可以從多個(gè)特征層中檢測到。

27.jpg

FoveaBox添加了2個(gè)子網(wǎng)絡(luò),一個(gè)子網(wǎng)絡(luò)預(yù)測分類,另一個(gè)子網(wǎng)絡(luò)預(yù)測bbox。

Object Fovea:

28.jpg

目標(biāo)的中央凹如上圖所示。目標(biāo)中央凹只編碼目標(biāo)對象存在的概率。為了確定位置,模型要預(yù)測每個(gè)潛在實(shí)例的邊界框。

FSAF、FCOS、FoveaBox的異同點(diǎn):

1.都利用FPN來進(jìn)行多尺度目標(biāo)檢測。

2.都將分類和回歸解耦成2個(gè)子網(wǎng)絡(luò)來處理。

3.都是通過密集預(yù)測進(jìn)行分類和回歸的。

4.FSAF和FCOS的回歸預(yù)測的是到4個(gè)邊界的距離,而FoveaBox的回歸預(yù)測的是一個(gè)坐標(biāo)轉(zhuǎn)換。

5.FSAF通過在線特征選擇的方式,選擇更加合適的特征來提升性能,F(xiàn)COS通過center-ness分支剔除掉低質(zhì)量bbox來提升性能,F(xiàn)oveaBox通過只預(yù)測目標(biāo)中心區(qū)域來提升性能。

總結(jié):

1.各種方法的關(guān)鍵在于gt如何定義

2.主要是基于關(guān)鍵點(diǎn)檢測的方法和密集預(yù)測的方法來做Anchor-Free

3.本質(zhì)上是將基于anchor轉(zhuǎn)換成了基于point/region

下一期我們詳細(xì)說說商湯的《CentripetalNet: Pursuing High-quality Keypoint Pairs for Object Detection》,基于向心偏移的anchor-free目標(biāo)檢測網(wǎng)絡(luò)centripetalnet,為基于關(guān)鍵點(diǎn)的目標(biāo)檢測方法研究帶來了新思路。

29.png

*博客內(nèi)容為網(wǎng)友個(gè)人發(fā)布,僅代表博主個(gè)人觀點(diǎn),如有侵權(quán)請聯(lián)系工作人員刪除。

隔離器相關(guān)文章:隔離器原理
電流傳感器相關(guān)文章:電流傳感器原理


關(guān)鍵詞: AI

相關(guān)推薦

技術(shù)專區(qū)

關(guān)閉