CVPR 2020 |用于3D目標(biāo)檢測的層級圖網(wǎng)絡(luò)
GU-net
本文設(shè)計了一個下采樣模塊,并將其重復(fù)堆疊 4 次以形成下采樣路徑,而將一個上采樣模塊重復(fù)堆疊兩次以構(gòu)成上采樣方式。類似 FPN、GU-net 生成三張點特征圖的特征金字塔。下采樣使用的是 FPS,然后通過KNN構(gòu)建局部區(qū)域,再使用 SA-GConv 更新特征,上采樣模塊的過程與下采樣模塊的過程相反,主要由 SA-GConv 執(zhí)行。
候選生成器
GU-net 生成了包含多級語義的三張點特征圖。一些先前的方法(如 VoteNet)僅使用一個特征圖進(jìn)行目標(biāo)預(yù)測。即使通過在上采樣過程中融合較低層的特征來計算較高層的特征,由于不同層的特征提供了各種語義,因此將多層特征一起用于候選生成會更加有益。本文提出了一種候選生成器,以改進(jìn)的投票模塊作為主要結(jié)構(gòu)來預(yù)測對象中心,該模型將多級特征轉(zhuǎn)換為相同的特征空間。接下來為了聚合特征,通過 FPS 保留 Np 的投票,該做法與 VoteNet 類似,從而融合多級特征以預(yù)測邊界框及其類別。
候選推理模塊
通過以上幾步,多層局部的語義信息已經(jīng)被很好的捕捉到了,但全局信息還沒有很好的學(xué)到,或者說可能有些目標(biāo)在點云中只體現(xiàn)出很小的一部分表面的點,在這樣少的信息下很難正確的將其識別出來。其推理過程為:
其中 Hp 表示候選特征 tensor,P 表示候選的相對位置
論文實驗
本文在 SUN RGB-D 和 ScanNet-V2 兩個數(shù)據(jù)集上進(jìn)行了實驗。
此外,本文還進(jìn)行了消融實驗以證明各?斓挠行。
結(jié)論
本文提出了一種新穎的 HGNet 框架,該框架通過層級圖建模學(xué)習(xí)語義。
具體來說,作者提出了一種新穎且輕巧的形狀注意圖卷積來捕獲局部形狀語義,該語義聚合了點的相對幾何位置的特征; SA-GConv 和 SA-DeGConv 構(gòu)建了 GU-net,生成了包含多級語義的特征金字塔。要素金字塔投票的點將位于相應(yīng)的對象中心,并且進(jìn)一步聚合多級語義以生成候選。然后使用 ProRe 模塊在候選之間合并和傳播特征,從而利用全局場景語義來提高檢測性能。最后,對邊界框和類別進(jìn)行了預(yù)測。

請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個字
最新活動更多
推薦專題
- 1 UALink規(guī)范發(fā)布:挑戰(zhàn)英偉達(dá)AI統(tǒng)治的開始
- 2 北電數(shù)智主辦酒仙橋論壇,探索AI產(chǎn)業(yè)發(fā)展新路徑
- 3 “AI寒武紀(jì)”爆發(fā)至今,五類新物種登上歷史舞臺
- 4 降薪、加班、裁員三重暴擊,“AI四小龍”已折戟兩家
- 5 國產(chǎn)智駕迎戰(zhàn)特斯拉FSD,AI含量差幾何?
- 6 光計算迎來商業(yè)化突破,但落地仍需時間
- 7 東陽光:2024年扭虧、一季度凈利大增,液冷疊加具身智能打開成長空間
- 8 地平線自動駕駛方案解讀
- 9 封殺AI“照騙”,“淘寶們”終于不忍了?
- 10 優(yōu)必選:營收大增主靠小件,虧損繼續(xù)又逢關(guān)稅,能否乘機器人東風(fēng)翻身?