CenterNet網(wǎng)絡精讀與分析

近期CVPR2019的最新論文CenterNet在一篇文章中就提出了一個能夠解決目標檢測、姿態(tài)檢測、3D單目檢測的掉炸天網(wǎng)絡熏挎,一時間激起來了千層波浪擅威。這幾乎是未來目標檢測一個全新的領(lǐng)域,或者說擯棄了一樣老舊的目標檢測思路景描,開始了一個新的紀元。

讓我們來看一下這篇論文能夠做的:


image.png

首先速度很快。

image.png

幾乎是一個全能的網(wǎng)絡芒粹。但是在這絢麗的表面背后,我們來窺探一下它的內(nèi)部運作原理大溜,以及它存在的問題化漆,從而才可以從某些方面去完善優(yōu)化它。

原理

One stage detectors 在圖像上滑動復雜排列的可能bbox(即錨點),然后直接對框進行分類钦奋,而不會指定框中內(nèi)容座云。
Two-stage detectors 對每個潛在框重新計算圖像特征,然后將那些特征進行分類锨苏。
這是以往的目標檢測方法所采用的理論方法基礎疙教。無論是one stage 或者是 two stage 都存在一個后處理:后處理,即 NMS(非極大值抑制)伞租,通過計算Bbox間的IOU來刪除同個目標的重復檢測框贞谓。這種后處理很難區(qū)分和訓練,因此現(xiàn)有大多檢測器都不是端到端可訓練的葵诈。

本文通過目標中心點來呈現(xiàn)目標(見圖2)裸弦,然后在中心點位置回歸出目標的一些屬性,例如:size, dimension, 3D extent, orientation, pose作喘。 而目標檢測問題變成了一個標準的關(guān)鍵點估計問題理疙。我們僅僅將圖像傳入全卷積網(wǎng)絡,得到一個熱力圖泞坦,熱力圖峰值點即中心點窖贤,每個特征圖的峰值點位置預測了目標的寬高信息。

模型訓練采用標準的監(jiān)督學習,推理僅僅是單個前向傳播網(wǎng)絡赃梧,不存在NMS這類后處理滤蝠。不存在后處理是CenterNet類似網(wǎng)絡一個巨大特點,也是一個非常好的優(yōu)點授嘀,現(xiàn)有的目標檢測方法在nms階段就得浪費大量的時間物咳。

image.png

在于此同時,我們也感到很熟悉蹄皱,這種方法與現(xiàn)在的姿態(tài)檢測是不是非常相似览闰?

  • 都是通過預測熱力圖峰值點來對關(guān)鍵點進行定位;
  • 都是端到端的熱力圖峰值點訓練方式

但是二者也存在這很大的差別巷折,最重要的一點压鉴,CenterNet將這種方法進行了橫向與縱向的擴張。這一點尤為重要盔几,泛華之后的應用帶來的是不同領(lǐng)域的嘗試和結(jié)合晴弃,結(jié)果證明,他們的嘗試取得了非常不錯的結(jié)果逊拍。

對于3D BBox檢測上鞠,我們直接回歸得到目標的深度信息,3D框的尺寸芯丧,目標朝向芍阎;

對于人姿態(tài)估計,我們將關(guān)節(jié)點(2D joint)位置作為中心點的偏移量缨恒,直接在中心點位置回歸出這些偏移量的值谴咸。

image.png

關(guān)于3D關(guān)鍵點檢測,這個圖非常重要骗露,它演示了像物體的長岭佳、寬、高是如何預測的萧锉。

除此之外珊随,我們額外關(guān)注一下這個網(wǎng)絡的loss設計:

image.png

其中Y_xyc是一個高斯核,對應的是GT和預測值的誤差柿隙。

推理

網(wǎng)絡的實際輸出是熱力圖上的每個類別的峰值叶洞,這與PoseNet以及其他所有的姿態(tài)檢測網(wǎng)絡類似,但當我們知道了這些峰值之后禀崖,如何回歸出需要的box呢衩辟?
做法是將熱力圖上的所有響應點與其連接的8個臨近點進行比較,如果該點響應值大于或等于其八個臨近點值則保留波附,最后我們保留所有滿足之前要求的前100個峰值點艺晴。

3d檢測部分:

3D檢測是對每個目標進行3維bbox估計昼钻,每個中心點需要3個附加信息:depth, 3D dimension, orientation财饥。我們?yōu)槊總€信息分別添加head.
其中depth非常難回歸换吧,需要做一些處理轉(zhuǎn)換,眾所周知钥星,當我們知道了物體的距離,3D尺寸满着,旋轉(zhuǎn)角度之后谦炒,物體的空間位置也就隨之確定了。這對于從單目中直接得到物體的真實距離非常重要7缋宁改!

更多

關(guān)于更多的論文解讀與討論,可以登錄Mana未來社區(qū):
http://ft.manaai.cn

進行關(guān)注魂莫,國內(nèi)最大的AI代碼市場已經(jīng)上線:
http://manaai.cn
歡迎大家的關(guān)注

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末还蹲,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子耙考,更是在濱河造成了極大的恐慌谜喊,老刑警劉巖,帶你破解...
    沈念sama閱讀 217,277評論 6 503
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件倦始,死亡現(xiàn)場離奇詭異斗遏,居然都是意外死亡,警方通過查閱死者的電腦和手機鞋邑,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,689評論 3 393
  • 文/潘曉璐 我一進店門诵次,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人枚碗,你說我怎么就攤上這事逾一。” “怎么了肮雨?”我有些...
    開封第一講書人閱讀 163,624評論 0 353
  • 文/不壞的土叔 我叫張陵遵堵,是天一觀的道長。 經(jīng)常有香客問我酷含,道長鄙早,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,356評論 1 293
  • 正文 為了忘掉前任椅亚,我火速辦了婚禮限番,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘呀舔。我一直安慰自己弥虐,他們只是感情好扩灯,可當我...
    茶點故事閱讀 67,402評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著霜瘪,像睡著了一般珠插。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上颖对,一...
    開封第一講書人閱讀 51,292評論 1 301
  • 那天捻撑,我揣著相機與錄音,去河邊找鬼缤底。 笑死顾患,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的个唧。 我是一名探鬼主播江解,決...
    沈念sama閱讀 40,135評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼徙歼!你這毒婦竟也來了犁河?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 38,992評論 0 275
  • 序言:老撾萬榮一對情侶失蹤魄梯,失蹤者是張志新(化名)和其女友劉穎桨螺,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體画恰,經(jīng)...
    沈念sama閱讀 45,429評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡彭谁,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,636評論 3 334
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了允扇。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片缠局。...
    茶點故事閱讀 39,785評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖考润,靈堂內(nèi)的尸體忽然破棺而出狭园,到底是詐尸還是另有隱情,我是刑警寧澤糊治,帶...
    沈念sama閱讀 35,492評論 5 345
  • 正文 年R本政府宣布唱矛,位于F島的核電站,受9級特大地震影響井辜,放射性物質(zhì)發(fā)生泄漏绎谦。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,092評論 3 328
  • 文/蒙蒙 一粥脚、第九天 我趴在偏房一處隱蔽的房頂上張望窃肠。 院中可真熱鬧,春花似錦刷允、人聲如沸冤留。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,723評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽纤怒。三九已至糯而,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間泊窘,已是汗流浹背熄驼。 一陣腳步聲響...
    開封第一講書人閱讀 32,858評論 1 269
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留州既,地道東北人谜洽。 一個月前我還...
    沈念sama閱讀 47,891評論 2 370
  • 正文 我出身青樓,卻偏偏與公主長得像吴叶,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子序臂,可洞房花燭夜當晚...
    茶點故事閱讀 44,713評論 2 354

推薦閱讀更多精彩內(nèi)容