ViP3D: 通過(guò)3D智體query實(shí)現(xiàn)端到端視覺(jué)軌跡預(yù)測(cè)
arXiv論文“ViP3D: End-to-end Visual Trajectory Prediction via 3D Agent Queries“,22年8月2日上傳,清華、上海(姚)期智研究院、CMU、復(fù)旦、理想汽車和MIT等的聯(lián)合工作。
現(xiàn)有的自主駕駛流水線將感知模塊與預(yù)測(cè)模塊分開(kāi)。這兩個(gè)模塊通過(guò)人工選擇的特征進(jìn)行通信,如智體框和軌跡作為接口。由于這種分離,預(yù)測(cè)模塊僅從感知模塊接收部分信息。更糟糕的是,來(lái)自感知模塊的錯(cuò)誤可能會(huì)傳播和累積,從而對(duì)預(yù)測(cè)結(jié)果產(chǎn)生不利影響。
這項(xiàng)工作提出ViP3D,一種視覺(jué)軌跡預(yù)測(cè)流水線,利用原始視頻的豐富信息預(yù)測(cè)場(chǎng)景中智體的未來(lái)軌跡。ViP3D在整個(gè)流水線中使用稀疏智體query,使其完全可微分和可解釋。此外,提出一種新的端到端視覺(jué)軌跡預(yù)測(cè)任務(wù)的評(píng)估指標(biāo),端到端預(yù)測(cè)精度(EPA,End-to-end Prediction Accuracy),其在綜合考慮感知和預(yù)測(cè)精度的同時(shí),對(duì)預(yù)測(cè)軌跡與地面真實(shí)軌跡進(jìn)行評(píng)分。
如圖是傳統(tǒng)多步級(jí)聯(lián)流水線與ViP3D的比較:傳統(tǒng)的流水線涉及多個(gè)不可微模塊,例如檢測(cè)、跟蹤和預(yù)測(cè);ViP3D將多視圖視頻作為輸入,以端到端的方式生成預(yù)測(cè)軌跡,可有效利用視覺(jué)信息,比如車輛轉(zhuǎn)向信號(hào)。
ViP3D旨在以端到端的方式解決原始視頻的軌跡預(yù)測(cè)問(wèn)題。具體而言,給定多視圖視頻和高清地圖,ViP3D預(yù)測(cè)場(chǎng)景中所有智體的未來(lái)軌跡。
ViP3D的總體流程如圖所示:首先,基于查詢的跟蹤器處理來(lái)自周圍攝像機(jī)的多視圖視頻,獲得有視覺(jué)特征所跟蹤智體的query。智體query中的視覺(jué)特征,捕獲智體的運(yùn)動(dòng)動(dòng)力學(xué)和視覺(jué)特征,以及智體之間的關(guān)系。之后,軌跡預(yù)測(cè)器將跟蹤智體的query作為輸入,并與HD地圖特征相關(guān)聯(lián),最后輸出預(yù)測(cè)的軌跡。
基于query的跟蹤器從環(huán)繞攝像機(jī)的原始視頻中提取視覺(jué)特征。具體而言,對(duì)于每一幀,按照DETR3D提取圖像特征。對(duì)于時(shí)域特征聚合,按照MOTR(“Motr: End-to-end multiple-object tracking with transformer“. arXiv 2105.03247, 2021)設(shè)計(jì)了一個(gè)基于query的跟蹤器,包括兩個(gè)關(guān)鍵步驟:query特征更新和query監(jiān)督。智體query會(huì)隨時(shí)間更新,建模智體的運(yùn)動(dòng)動(dòng)力學(xué)。
大多數(shù)現(xiàn)有的軌跡預(yù)測(cè)方法可分為三個(gè)部分:智體編碼、地圖編碼和軌跡解碼。在基于query的跟蹤之后,獲得被跟蹤智體的query,該query可以被視為通過(guò)智體編碼獲得的智體特征。因此,剩下的任務(wù)是地圖編碼和軌跡解碼。
分別將預(yù)測(cè)和真值智體表示為無(wú)序集S?和S,其中每個(gè)智體由當(dāng)前時(shí)間步的智體坐標(biāo)和K個(gè)可能的未來(lái)軌跡表示。對(duì)于每個(gè)智體類型c,計(jì)算Sc?和Sc之間的預(yù)測(cè)精度。將預(yù)測(cè)智體和真值智體之間的成本定義為:
這樣Sc?和Sc之間的EPA定義為:
實(shí)驗(yàn)結(jié)果如下:
注:這個(gè)目標(biāo)渲染做的不錯(cuò)。