小議車輛環境視覺基礎前視感知

1. 前言

無論是AD/ADAS還是智能網聯車,前視感知都是其最基礎和重要的能力之一。自動駕駛(AD)是前幾年的熱門話題。今天雖然稍微降溫下來一些,但仍是大家關注的重點之一,畢竟它是人類長久以來的夢想之一。衆所周知,美國汽車工程師學會(SAE)將自動駕駛分爲 L0~L5共六個級別。其中L3及以上允許由系統在限定或不限定條件下完成所有的駕駛操作;而L2及以下還是需要由人類駕駛員一直保持駕駛狀態,因此大多還屬於高級駕駛輔助系統(ADAS),如車道偏離預警(LDW),前碰撞預警(FCW),自適應巡航(ACC),緊急自動剎車(AEB),車道保持輔助(LKA),交通標誌識別(TSR),自動泊車(AP)等。和大多數智能機器人一樣,自動駕駛的處理流程可分爲三個階段:感知、決策、執行。因此一旦感知出了問題,那後面基本就涼了。類似地,ADAS或AR導航也強依賴於對環境的感知。沒有準確且實時的感知能力,上層做的再炫酷也容易成爲雞肋。ADAS和AD間的界線並沒那麼清晰,前者可看作到後者的過渡產品,因此很多技術是通用的。前視感知是個非常大的話題,因此本文主要聚焦在一些最爲基礎和通用的前視感知能力上。

2. 業界

本節我們從工業界和學術界兩個方面簡要聊下業界的相關情況。它們各有優缺點,學術界涌現出更前沿更先進的方法,且指標明確,易於定量比較,但方法往往專注於單點,且對實際產品中的各種約束(如計算資源)考慮不多;而工業界直接面對產品,更多地考慮實用性和整體性。但採用的指標、數據不透明,難以衡量和比較。只有全方面的瞭解,通過產學研的加速融合,才能打造更加完善、更好用戶體驗的產品。

ADAS有着幾十年的發展歷史。國內外都有一大批優秀的廠商。這幾年,隨着國家駕駛安全政策的推動和自動駕駛技術受到熱捧,該領域出現了快速的增長。從老牌勁旅Bosch、Continental、Aptiv,Mobileye等,到一批相對年輕但很有競爭力的公司如Maxieye、Minieye、魔視、極目、縱目、Nauto等,這是一個既成熟,又充滿機遇的市場。根據中投顧問的《2017-2021年中國汽車高級駕駛輔助系統(ADAS)市場深度調研及投資前景預測報告》,ADAS年複合增長率將達35%,2020年中國市場可實現近800億市場空間。近幾年,車載AR導航將傳統的ADAS功能與導航功能、AR技術及HUD進行了融合,帶來了更直觀和人性化的用戶體驗,成爲了市場的熱點。在實現方式上,各家在傳感器配置上也各有不同,有攝像頭、毫米波雷達、激光雷達等。其中,基於攝像頭的視覺方案由於其成本可控、算法成熟等優點,使用最爲廣泛。其中的主要代表如Mobileye和Tesla Autopilot都是主要基於視覺的方案。
在這裏插入圖片描述
圖 1 Tesla Autopilot (來自https://www.youtube.com/watch?v=24dRkHdpEPo)
在這裏插入圖片描述
圖 2 Mobileye (來自https://www.mobileye.com/our-technology/)

雖然ADAS細分功能衆多,但很多功能功能(LDW,FCW,LKA,ACC等)都依賴於對前方環境中幾個基本對象的檢測和識別,即車道線、物體(包括車輛、行人、障礙物、交通燈、交通標識等)、可行駛區域,因此本文也會主要聚集在這幾類對象的檢測識別上。在準確率上,各家的產品往往很難量化及橫向比較,儘管大家的宣傳中常會出現“準確率>XX%”或者“誤報率/漏報率<X%”這樣的提法。但事實上由於指標定義和數據集的差異,很難做到“apples to apples”的對比。有些比較實在的會採用相對指標,比如與行業標杆(如Mobileye)在同樣的數據集上做橫向對比。這樣會更有說服力。但因爲所用數據集未公開,別人也很難嚴格橫向比較。在性能上,由於ADAS對檢測的實時性要求非常高,所以很多至少會做到15 FPS以上。隨着深度學習的興起,越來越多的ADAS廠商也將基於傳統CV的方案逐漸過渡爲基於深度學習的方案。深度神經網絡(DNN)的推理需要很大的計算量,而ADAS不像高級自動駕駛平臺配備強大的計算平臺,因此業界的趨勢之一是利用專用硬件(如FPGA或ASIC芯片)來進行加速。

在學術界,自動駕駛一直是經久不衰的熱點之一。這些方法上的創新很多同樣也可以用於ADAS和AR導航中。我們知道,2012年以來深度學習的快速發展使其成爲機器學習中的絕對主流。基於深度學習的方法同樣也給自動駕駛帶來了巨大變革。基於傳統CV算法的方法在泛化能力上容易遇到瓶頸。經常是在一段路段調優跑溜後,換一段路又需要大量調參。當然,基於深度學習的方法也無法完全避免這個問題,但可以說是大大緩解了。學術界的優點就是較爲透明公開、且容易對比。新的方法就是需要在與其它方法比較中才能證明其優異,因此歷史上通過競賽的方式來推動發展的例子不在少數。2004年開始,由DARPA主辦的幾場無人車挑戰賽開啓了無人車的新時代。在深度學習時代,各種針對路面環境檢測識別的榜單就如同ImageNet一樣,催生出一大批新穎的方法。其中針對車道線、物體和可行駛區域的比較典型的有:

  • KITTI:2013年由德國卡爾斯魯厄理工學院和豐田美國技術研究院聯合創辦,是一套非常全面的算法評測數據集。其中覆蓋了2D、3D物體檢測,物體跟蹤,語義分割、深度估計等多種任務。但其數據集數量在現在看起來不算多,如物體檢測數據集訓練集和測試集總共1W5張左右 ,車道檢測就比較尷尬了,只有幾百張。
  • CVPR 2017 TuSimple Competitions:2017年圖森未來主持的挑戰賽,分車道線檢測和速度估計兩個任務。其中車道線檢測數據集包含了幾千張主要是高速上的數據。雖然量不多,但因爲和物體檢測這類通用任務相比,針對車道線檢測的競賽很少,所以它至今在很多車道線檢測的論文上還會被當成量化比較的重要參考。
  • CVPR 2018 WAD:由Berkeley DeepDrive主持,包含三項賽事:道路物體檢測,可行駛區域分割和語義分割的域適應。它基於BDD100K數據集。這是一個在數量和多樣性上都非常不錯的用於自動駕駛的數據集。其中用於物體檢測和可行駛區域分割的數據集共有10W張左右,其中訓練集和驗證集有約8W張 。
  • Cityscapes:針對道路環境的經典語義分割數據集,同時也提供了語義分割、實例分割和全景分割任務的榜單。數據集採自50個城市,包含了5K張精細標註圖片和2W張較粗糙標註圖片。考慮到語義標註的成本相對大,這個數量已經算比較大了。

其它的榜單還有很多,無法一一列舉。雖然由於大多榜單隻關注準確率導致其模型很難直接落到產品中,但其中確實也出現了非常多精巧的方法與創新的想法,爲產品落地提供了有價值的參考。關於具體的方法我們留到後面專門章節進行討論。

還有一些介於工業界與學術界之間的工作,它們將學術界的成果向產品逐漸轉化,提供了參考實現。比較典型的有開源自動駕駛項目Apollo和Autoware。因爲它們主要面向無人駕駛,所以會除了攝像頭之外,還會考慮激光雷達、毫米波雷達、高精地圖等信息。由於本文的scope,這裏只關注基於攝像頭的對基礎對象的檢測。Apollo 2.5中採用的是一個多任務網絡檢測車道線與物體(之前讀代碼的一些筆記:自動駕駛平臺Apollo 2.5閱讀手記:perception模塊之camera detector ),對於車道線模型會輸出像素級的分割結果,然後通過後處理得到車道線實例及結構化信息(相關代碼閱讀筆記:自動駕駛平臺Apollo 3.0閱讀手記:perception模塊之lane post processing);物體檢測是基於Yolo設計的Yolo 3D,除了輸出傳統的2D邊界框,還會輸出3D物體尺寸及偏轉角。版本3.0(官方介紹:perception_apollo_3.0)中加入了whole lane line特性,提供更加長距的車道線檢測。它由一個單獨的網絡實現。3.5中將物體與車道線檢測網絡徹底分離,車道線模型稱爲denseline。最新的5.0(官方介紹:perception_apollo_5.0)中又引入DarkSCNN模型,它基於Yolo中的backbone Darknet,並引入了Spatial CNN(後面再介紹),同時該網絡中還加入了對滅點的檢測。Autoware中車道線用的是傳統CV的方法,物體檢測基於攝像頭部分使用的是SSD和YOLO等基於深度學習的方法(官方介紹:Overview)。

3. 方法

我們知道,深度學習有三大基石:數據、算法和算力。對於基礎前視感知場景,我們也從這三個維度來聊一下。在此之前,我們先列下本文關注的單目基礎感知主要流程:

在這裏插入圖片描述
圖 3 基礎前視感知簡要流程

從圖片流輸入,大致經歷預處理、檢測模型推理和後處理三個階段:

  • 預處理:這一階段主要是做必要的數據處理,爲後面的檢測準確好數據。如一些攝像頭進來的視頻流是YUV格式,而深度神經網絡模型輸入多是RGB格式,需要進行轉換。此外,很多模型會要求輸入數據作歸一化。另外,爲了減少計算量,一般還會對輸入的圖像進行縮放和ROI的提取。
  • 檢測模型推理:這一階段主要是做深度神經網絡的推理。對於要同時完成多個任務的場景,我們一般會使用多任務網絡。即每個任務對應一個網絡分支輸出,它們共享用於特徵提取的backbone(按經驗很多時候backbone會佔大部分的計算量)。近年學術界也出現一些對不同類型任務比較通用的backbone結構。
  • 後處理:這一階段是將前面推理得到的結果進行進一步的處理,以傳到後面的決策或展示模塊。常見的對於車道線需要濾除噪點,聚類,曲線擬合,濾波(如Kalman filter)等;對物體檢測常見的有非極大值抑制(NMS)和跟蹤等;對可行駛區域,需要將分割結果轉爲多邊形並確定其位置類別。

注意這裏只畫了簡化的部分流程。實際場景中,可能還需要考慮非常多其它元素,比如:

  • 相機標定(Camera calibration):我們在學車考“S彎”或者“單邊橋”等項目時,教練往往會告訴我們一些小技巧,如通過雨刮器的位置來估計輪胎的位置。這種技巧其實比較脆弱,因爲座椅的調整,人的高矮都會影響其精確度。那在ADAS/AD場景中如何告訴機器以高精度做這件事呢,就是通過相機校準。這本質是做圖像座標和世界座標之間的轉換。另外,有了校準參數,我們還可以用它做逆透視映射(IPM),消除透視帶來的影響,方便車道線檢測及物體跟蹤等模塊。
  • 光流(Optical flow):每一幀都檢測會帶來很大的計算開銷。有時我們會通過光流算法來計算圖像中像素點的瞬時速度,從而估計已檢測對象在當前幀的位置。這樣一方面能有效減少計算量,另一方面還能用於物體的跟蹤。
  • 滅點(Vanishing point):我們知道,由於透視關係,平行的線(如車道線)在遠處會交到一點,稱爲消失點或滅點。這個點對於車道檢測或最後的可視化都有幫助。在直線的情況下,我們可以通過車道線的交點來估計滅點,但如果車道線是不太規則的曲線,就比較麻煩,需要通過更復雜的方法進行估計。
  • 測距:不少ADAS功能中都需要確定前方物體的距離。常用毫米波和超聲波雷達做距離檢測。而在純視覺方案中,雙目方案是根據視差來估計距離,原理就像人的兩隻眼睛一樣。而對於單目方案就比較tricky一些,需要檢測物體後根據物體下邊界結合相機標定計算距離。現在雖然有基於單張圖像的深度估計方法,但那個本質上是靠的訓練所獲得的先驗,用作ADAS裏的FCW啥的感覺還是精度不太夠。
  • 3D姿態估計(3D pose estimation):高級點的前視感知對物體檢測除了邊界框,還會估計其姿態。這對動態障礙物的行爲預測非常有幫助。
  • 細粒度識別(Fine-grained recognition):對於一些識別的物體,如果它們的類別會影響到駕駛行爲(如交通燈、交通標誌、車道線等),則我們需要將檢測結果中相應部分拿出來進一步對其進行分類識別。
  • 決策和展示:所有的檢測都是爲了最後的決策和展示。如何自然地顯示(如通過AR展示的話如何與現實物體貼合),以及何時預警或介入控制都直接影響用戶體驗。

另外可能還需要檢測路面上的指示標記,以及對當前場景是否支持作檢測判斷等等。由於篇幅有限和使內容簡潔,這些本文都暫不涉及。檢測對象上本文主要關注車道線、物體和可行駛區域。

3.1 數據

我們知道,深度學習的最大優勢之一就是能對大量數據進行學習。這就意味它的效果很大程度上依賴於訓練的數據量,而對於汽車的前視感知更是如此。因爲汽車的環境是開放的,沒有充分而多樣的數據,模型便無法有效地泛化,那在各種corner case就可能出岔子。對於其它場景出岔也就出岔了,對AD或者ADAS來說那可能就危及生命安全了。數據集大體有兩類來源,一類是公開數據集;一類是自標數據集。它們各自有優缺點。

得益於自動駕駛領域的蓬勃發展,近年來出現了很多優質的公開數據集。

  • 包含車道線的主要有BDD100K,CULane,TuSimple Lane Challenge和LLAMAS等。
  • 包含物體檢測的太多了,貌似是個道路環境數據集就會有。如BDD100K,KITTI,Udacity Driving Dataset, Waymo Open Datasets等。
  • 包含可行駛區域的有BDD100K,KITTI等。雖然理論上語義分割的數據集(如Cityscapes)就包含了可行駛區域的標註,但比較理想的標註還應該區分當前車道和相鄰車道。

其它的還有不少數據集,網上有很多列表整理,這裏就不重複了。雖然這些數據集很豐富,但有時未必能直接用上。一方面是它們的標註之間有很大差異。其中一個差異點是標註格式,這個其實還好辦,腳本基本能搞定。比較麻煩的是有時候標註的規範和內容會有出入。以車道線爲例:有些是採用雙線標法(如BDD100K),有些是單線標法(如CULane,TuSimple Lane Challenge);有些是標有限條(如CULane),有些是有多少標多少(如BDD100K);有些對於虛線是像素級精確標註(ApolloScape),有些是會將它們“腦補”連起來(CULane);有些標了車道線類型(BDD100K),有些沒有標(CULane)。而對於車輛和行人來說,不同數據集有不同的細分類。但本着人家標註也不容易,能用上一點是一點的精神,可以儘可能地對它們進行轉化,使它們一致並滿足特定需求。舉例來說,BDD100K中是雙線標註,而其它多數是單線標。爲了統一,我們可以通過算法自動找到匹配的線並進行合併。自動合併效果如下:

在這裏插入圖片描述
在這裏插入圖片描述
圖 4 BDD100K數據集車道線標註自動轉換

公開的數據集雖然方便且量大,但往往沒法完全滿足需求。比如由於地域差異、攝像頭差異等會導致domain shift問題,另外有些針對性的case沒法覆蓋。公開數據集另一個問題是license。很多的公開數據集只能作研究用途,如果要商業用途是禁止或者需要專門再購買license的。因此,實際中往往還是需要請外包或自己標數據。

另外,爲了獲得更大量更多樣的數據,業界有一些常用方法和方向,如:

  • 數據增強(Data augmentation):最基本也很有效的擴充數據集手段之一,在車輛環境中尤爲重要。由於道路環境數據集需要多樣化,因此我們需要通過數據增強來模擬不同的光照、天氣、視角等變化。
  • 自動標註/輔助標註:雖然移動端上由於算力有限,我們只能犧牲準確率佈署輕量級的網絡,但我們可以訓練重量級的精度較高的網絡模型用於對數據進行自動標註。以下是一個重量級網絡(不是SOTA的)在BDD100K上訓練後的檢測效果。雖然不是十全十美,但在有些小目標上可能比老眼昏花的我還要標得湊合。就算無法完全替代人肉標註也可以作爲輔助有效減少人工。

在這裏插入圖片描述
圖 5 某重量級網絡在BDD100K數據集上檢測效果(左:檢測結果;右:Ground truth)

  • 仿真器:利用仿真器來幫助自動駕駛測試似乎已經是一個普遍性做法了。隨着3D圖形技術和硬件的飛速發展,今天仿真器中的渲染效果已相當逼真,已經不像當年賽車遊戲裏車後冒個煙還是“馬賽克”效果。因此,仿真器也有望用於產生可用於訓練的數據。
  • 生成對抗網絡(GAN):我們知道,GAN是最近幾年非常火熱的一個方向。GAN也在一些工作中用於訓練數據的生成。雖然目前很多時候是看demo各種牛,但實際跑的時候可能就不是很理想。但不可否認這是一個很有前途的方向,不少工作應用它來緩解數據多樣化需求的問題。

3.2 算法

針對前視感知中的幾類目標,算法是不同的。另一方面,我們知道深度學習的視覺領域研究比較多的任務是:圖片識別、物體檢測、圖像分割(包括語義分割、實例分割、全景分割)。那麼問題來了,如何將對現有任務的方法充分應用來滿足前視感知的需求?如果實在不合適如何調整?

3.2.1 車道線

首先是車道線檢測,這可能是幾類檢測目標中最特殊的,所以佔的筆墨也會相對多些。它的特點是形狀狹長(可能跨越大半張圖片),並且形態多變(可能是直線也可能是曲線,還可能交叉等),容易與路面標識混淆,另外還需要區分實例。現有物體檢測的方法不太適合這種形狀的東西。我們知道,在深度學習佔領視覺領域前,車道線檢測多采用傳統CV的方法。Udacity(其聯合創始人Sebastian Thrun是自動駕駛界大神)上有一個自動駕駛課程。其中有作業就是車道線檢測,因此網上有很多這個作業的實現。其中比較關鍵的幾步是通過邊緣檢測算法(如Canny,Sobel算子)得到邊緣,然後通過Hough transform檢測直線(如果假設車道線爲直線),或者經過IPM得到鳥瞰視圖後通過滑窗搜索得到車道線上的像素點,最後多項式曲線擬合輸出。這裏邊幾乎每一步都有不少參數,而且各步相互影響,如果場景很多樣化的話調參就可能會比較酸爽,另一方面它對於車道線不完整的情況(如因遮擋或磨損)表現不好。因此,這已經不是目前的主流,後面業界逐漸過渡到基於深度學習的方法。

2015年,深度學習風頭正勁,Stanford、Twitter等機構聯合發表的論文討論了將CNN應用到高速環境的車道線和車輛檢測中[1]。它使用當時物體檢測的方法[2]來檢測車道線。因爲車道線很長條,因此被分成多個線段,每個線段被當成物體來檢測。最後通過DBSCAN進行聚類得到車道線實例。同期另外一條思路是將車道線檢測當作語義分割任務。當時語義分割領域有了FCN[3]、SegNet[4]和DeepLab[5]等早期經典網絡。結合一些包含車道線標註的語義分割數據集便可以進行車道線檢測。如論文[6]試圖將包括車道線在內的多種檢測任務在分割任務中一把搞定。然而故事還遠沒有結束,這裏還存在以下兩個比較大的挑戰,接下去幾年的工作也是主要圍繞這兩點來展開:

  • 繁瑣的後處理:現實中我們爲了後面的決策還需要知道哪條是當前車所在車道(Ego lane)的左、右車道線和相鄰車道的車道線。另外,因爲車道線往往不完整,因此還需要得到車道線的結構化表示(如多項式或樣條曲線)以便做插值。這樣,單就語義分割的結果還不夠。以往常見的做法是將分割結果進行聚類得到實例,然後通過一些後處理判斷其是哪條車道。另外,爲了得到結構化表示還需要對這些點進行多項式擬合等操作。理想的方法是簡化或完全去除這些後處理,實現真正意義上end-to-end的檢測。
  • 複雜的環境:路面環境複雜常常導致圖像中的車道線殘缺不全。如天氣因素,其它車輛遮擋,陰影和光照,磨損等等。另外的一個比較大的干擾來自於地面上的箭頭指示和漢字,僅看局部圖像的話人也難以區分。因此如果無法有效利用全局上下文信息很難對它們進行排除。對這些因素做到足夠魯棒是通往實用產品的必要條件。

來自三星的論文[7]將車左右兩條車道線作爲兩個類別(加上背景共三類),從而直接通過神經網絡來學習,相當於做了實例分割,從而簡化了後處理。

2017年TuSimple主辦了車道線檢測競賽,炸出不少好的方法,同時也成爲了車道線檢測的重要benchmark之一。第一名來自香港中文大學,它也是基於語義分割來做,並針對車道線這種狹長的物體提出了Spatial CNN(SCNN)[8]來替代MRF/CRF來對空間關係進行建模。另一個比較有意思的點是當時競賽提供的數據集才幾千張(標註圖片約3.6K),因此數據可能會成爲主要瓶頸之一,於是他們整了一個大規模的車道線的數據集CULane。該數據集共有13W多張。它比較貼近現實情況,涵蓋了白天、晚上、擁堵、陰影、光照過亮等9種場景。對於車道線的實例區分問題,SCNN由於限定最多檢測4條車道線,因此它可以把4條車道線當4類物體來檢測。同時,網絡還有一個專門的分支用於預測對應的車道線是否存在。這樣便不需要聚類來提取實例。當時的第二名來自佐治亞理工(Georgia Institute of Technology) 等機構。他們提出的方法[9]可以解決只能處理有限車道線的問題。它利用像素對之間的關係,通過對目標函數的巧妙構造,讓神經網絡學習像素的聚類信息。並且可以拓展到(理論上)無窮實例的場景。

2017年韓國KAIST和三星提出了VPGNet[10]。它是一個多任務網絡,其中一個分支用於預測滅點,它可以引導車道線的檢測。這在一些惡劣的天氣下可以有比較大的幫助。但這需要額外標註的數據集。論文中提到他們建立了自己的數據集但沒有公開。

2018年,魯汶大學(KU Leuven)的論文提出LaneNet[11],它將車道線檢測作爲一個實例分割問題。以前很多方法對於提取車道線實例是用聚類,而對於車道線這種狹長的物體很難定義一個好的距離測度用於聚類。這篇論文的最大特色就是在傳統語義分割分支外還加了一個pixel embedding分支,用於對輸入圖像中的每個點得到其N維的embedding,這個分支是基於其實例信息訓練的。語義分割輸出的像素結合pixel embedding信息,作聚類後便可得到車道線的實例信息,最後通過多項式擬合輸出。魯汶大學這個團隊次年在論文[12]中把預測曲線與ground truth曲線間的面積作爲損失函數,將擬合改造成可微分操作,從而讓神經網絡來學習擬合曲線的參數。前面LaneNet這篇論文另一個比較有特色的點是H-Net。IPM有利於車道線的多項式擬合。因爲大多數彎曲的車道線在鳥瞰視圖下用二次曲線就夠了,但在透視視圖下卻需要更高階曲線才能擬合。而這個變換的參數一般需要通過相機標定。但是這個參數可能根據地形、坡道因素不同。因此最好可以根據輸入動態調整。H-Net採用通過神經網絡來預測的方式。這條思路上類似的工作還有來自2018年GM的3D-LaneNet[13]。該方法以end-to-end方式直接預測3D的車道線。網絡採用dual-pathway結構。一條對應普通透視圖,估計逆透視變換參數。該參數結合前面的feature map與另一條對應鳥瞰視角的網絡中feature map結合,最終輸出3D車道線。不過畢竟帶3D車道線標註的數據集不好弄,於是他們自己搞了個高速場景下的合成數據集作了實驗。因此該方法在真實場景下的效果還需要進一步驗證。

我們知道,注意力(Attention)機制是深度學習這幾年很火的主題,它在視覺領域也有不少的應用。今年由香港中文大學等機構發表的論文[14]提出了Self Attention Distillation(SAD)方法。它基於注意力蒸餾(Attention distillation)的思想,將之改造爲自蒸餾,從而不依賴傳統知識蒸餾中的teacher model。網絡中後面的層的feature map(具有更豐富上下文信息)作爲監督信息幫助前面的層訓練。前面的層學到更好的表徵後又會改善後面的層,構成良性循環。

3.2.2 道路物體

然後是物體檢測,這塊的算法可以說是相當豐富。因爲物體檢測的應用範圍非常廣,因此它幾乎伴隨着計算機視覺領域的發展。相關的survey很多(如[15],[16]等)。深度學習興起後,一大波基於深度神經網絡的物體檢測算法被提出。SOTA以極快的速度被刷新。從two-stage方法到輕量的one-stage方法,從anchor-based方法到近年很火的anchor-free方法,從手工設計到通過自動神經網絡架構搜索,琳琅滿目,相關的總結與整理也非常多。

對於道路環境來說,幾乎和通用物體檢測算法是通用的。如果要找些區別的話,可能汽車前視圖像中,由於透視關係,小物體會比較多。2018年CVPR WAD比賽其中有一項是道路環境物體檢測。第一名方案來自搜狗,根據網上介紹(給機器配上“眼睛”,搜狗斬獲CVPR WAD2018挑戰賽冠軍),其方案在Faster R-CNN的基礎上使用了CoupleNet,同時結合了rainbow concatenation。第二名方案來自北京大學和阿里巴巴,提出了CFENet[17]。經典的one-stage物體檢測網絡SSD在多個scale下的feature map進行預測,使得檢測對物體的scale變化更加魯棒。小目標主要是通過淺層的較大feature map來處理,但淺層特徵缺乏包含高層語義的信息會影響檢測效果。CFENet針對前視場景中小物體多的特點對SSD進行了改進,在backbone後接出的淺層上加入CFE和FFB網絡模塊增強淺層特徵檢測小目標的能力。

現實應用中,物體檢測模型的輸出還需要經過多步後續的處理。其中比較常見和重要的是NMS和跟蹤:

  • 神經網絡模型一般會輸出非常多的物體框的candidate,其中很多candidate是重疊的,而NMS的主要作用就是消除那些冗餘的框。這個算子很多的推理框架不支持或支持不好,所以一般會放到模型推理外面作爲後處理來做。在學術界NMS這幾年也出現了一些可以提高準確率的變體。
  • 跟蹤是理解物體行爲的重要一環。比如幀1有車A和車B,幀2有兩輛車,我們需要知道這兩輛車哪輛是A,哪輛是B,或都不是。只有找到每個物體時間維度上的變化,才能進一步做濾波,以及相應的分析。比較常見的多物體跟蹤方法是SORT(Simple Online and Realtime Tracking)框架[18],或許它的準確率不是那麼出衆,但綜合性能等因素後還是不錯的選擇,尤其是對於在線場景。結合通過CNN提取的外觀特徵(在DeepSORT[19]中採用)和Kalman filter預測的位置定義關聯度的metric,將幀間物體的跟蹤作爲二分圖匹配問題並通過經典的匈牙利算法求解。前後幀物體關聯後通過Kalman filter對狀態進行更新,可以有效消除檢測中的抖動。

3.2.3 可行駛區域

再來說下可行駛區域。開過車的同志們都知道咱們的很多路沒有那麼理想的車道線,甚至在大量非結構化道路上壓根兒就沒有車道線。在這些沒有車道線、或者車道線不清晰的地方,可行駛區域就可以派上用場。一般在可行駛區域中我們需要區分當前車道和其它車道,因爲該信息對後面的決策規劃非常有價值。

在這個任務上早期比較流行的榜單是KITTI的road/lane detection任務。很多論文都是拿它作benchmark,其榜單上有一些是有源碼的。不過那個數據量比較少,多樣化程度也不夠,要用它訓練得泛化能力很強實在比較勉強。

2018年CVPR WAD比賽中一個專項是可行駛區域檢測。所用的BDD100K數據量相比豐富得多。當時的冠軍方案是來自香港中文大學的IBN-PSANet。它的方案是結合了IBN-Net[20]和PSANet[21]。前者主要特色是結合了batch normalization(BN)和instance normalization(IN)。BN幾乎是現代CNN的標配。它主要用於解決covariate shift問題,提高訓練收斂速度;而IN可以讓學習到的特徵不太受像顏色、風格等外觀變化的影響。而結合了兩者的IBN可以吸收兩者的優點。而PSANet的特色主要是提出了PSA結構,它本質是一種注意力機制在視覺上的應用。對於每一個像素,網絡學習兩個attention mask,一個對應它對其它每個像素的影響,一個對應其它每個像素對它的影響,從而使得分割可以充分考慮全局上下文信息。

可行駛區域檢測中對於語義分割的輸出比較粗糙,且形式不易於後面模塊處理,因此還需要經過一些簡單的後處理。比如先聚類,再計算各類簇的凸包,最後通過這些多邊形的位置關係便可以確定它們是當前車道還是其它車道的可行駛區域。

值得一提的是,可行駛區域和車道線語義上是非常相關的,因此可以通過相互的幾何約束來提高準確率。業界也有不少這方面的嘗試,越來越多的深度神經網絡將它們進行融合。

3.3 優化

從算法到產品最大的鴻溝之一便是性能優化。移動端設備有限的算力正在與多樣化算法的算力需求形成矛盾。這在之前寫的文章《淺談端上智能之計算優化》中進行過初步的討論。對於像ADAS這樣的場景實時性尤其重要。我們可以從文中提及的幾個角度進行優化。

首先,在網絡設計上我們在backbone上可以選擇這幾年經典的輕量級網絡(如MobileNet系[22], [23],ShuffleNet系[24], [25],EfficientNet[26]等)。這些網絡一般在計算量上比重量級網絡有數量級上的減少,同時又可以保持準確率不損失太多。另一方面,對於多個檢測任務,由於輸入相同,我們一般會使用多分支的網絡結構。每個任務對應一個分支(head),它們共享同一個用於特徵提取的backbone。按經驗來說,這個backbone佔的計算一般會比較大,因此這樣可以節省下相當可觀的計算開銷。但是這樣的多任務多分支網絡會給訓練帶來困難。最理想的當然是有全標註的數據集,但這樣的數據集比較難獲得。對於這個問題,我們可以採取兩種方法:一種是如前面提的,靠重量級高準確率網絡自動標註。如訓練高準確率的物體檢測模型給已有車道線標註的數據集進行標註;另一種就是對帶特定標註的數據輸入,訓練對應的部分(backbone和相應的head)。

對於給定網絡結構,我們可以通過模型壓縮進一步減少計算量。因爲普遍認爲推理時不需要訓練時那樣複雜的模型和高的精度。模型壓縮有很多種方法,有量化、剪枝、知識蒸餾、低軼分解等等。常用的方法之一是量化。一般來說,將FP32轉爲FP16是一種既比較安全收益又比較大的做法,然而在一些低端設備上我們還需要作更低精度(8位或以下)的量化。這時就得花更多精力在準確率損失上了。量化又分爲post-training quantization和quantization-aware training。前者使用方便,不需要訓練環境,最多需要少量(幾百張)數據集作爲量化參數calibration之用,但缺點是會對準確率損失較大;而後者,需要在訓練時插入特殊的算子用於得到量化所用參數及模擬量化行爲。另一種常用的壓縮方法是網絡剪枝。根據網絡模型的敏感度分析,一些層稍作裁剪可能就會有大的準確率損失,而另一些層進行裁剪則準確率損失不大,甚至還會使準確率上升。這就給了我們簡化模型從而減少計算量的機會。低軼分解本質上是通過對矩陣的近似來減少矩陣運算的計算量。知識蒸餾是一種很有意思的方法,就像現實中的老師教學生,通過teacher model來幫助訓練student model。

網絡模型敲定後,就需要考慮性能優化。深度的優化是離不開硬件的考慮的。對於一些用於自動駕駛的計算平臺,可能直接就上像Nvidia的PX2這樣的高性能硬件平臺了。但對於普通車規硬件平臺,肯定是扛不住這種成本的。這些常規車機平臺中一些稍高端的會有幾百GFLOPS的GPU處理能力,或其它DSP,NPU等計算硬件。這裏我們一般會首選這些硬件做模型推理而非CPU。因爲如果將這些計算密集型任務往CPU放,會和系統中其它任務頻繁搶佔資源導致不穩定的體驗。而對於低端一些的平臺GPU基本只夠渲染,那隻能放到CPU上跑,一般會用上面提到的量化方法將模型轉爲8位整型,然後將推理綁定到固定的核上以防止影響其它任務。推理引擎有兩類選擇。對於一些有成熟推理引擎的硬件平臺,使用廠商的引擎(如Intel有OpenVINO,高通有SNPE)通常是一個方便快捷的選擇;還有一種方法就是用基於編譯器的推理引擎,典型的如TVM。它以offline的方式將網絡模型編譯成可執行文件並可進行自動的執行參數優化。至於哪個性能好,通常是case-by-case,需要嘗試。值得注意的是,上面選取的輕量型網絡一般是memory-bound的,因此優化時需要着力優化訪存。

如果平臺上有多種可以執行神經網絡算子的硬件,如CPU、GPU、NPU、DSP,那可以考慮通過異構調度來提高硬件利用率,從而達到性能的優化。現在業界已有不少的異構計算框架,如ONNXRuntime,Android NN runtime等。這裏面,最關鍵核心的問題在於調度。對於單個網絡模型而言,先要對網絡進行切分,然後分配到最合適的硬件上,然後在每個硬件上進行本地調度。難點在於這個調度是NP-hard的,意味着對於實際中大規模問題,不可能在合理時間找到最優解,而要找到儘可能優的近似解是門大學問。業界出現了大量的方法,如精確算法、基於啓發式策略、元啓發式搜索和機器學習的方法。對於前視感知任務中的多分支模型,一個最簡單而有效的做法就是將backbone以及各個head的分支作爲子圖進行切分和調度。如果要得到更優的調度,則可以進一步嘗試基於搜索和學習的方式。

4. 小結

前視感知領域是一個小打小鬧容易但做好非常難的東西。它需要長期的沉澱才能構建起核心競爭力和技術壁壘。我們看到今天行業龍頭Mobileye獨領風騷,但少有人看到它在早期的執着。Mobileye創立於1999年,但到2007年纔開始盈利。類似的還有谷歌的無人駕駛車(差不多10年了),波士頓動力的機器人(貌似27年了),還有許許多多這樣“耐得住寂寞”的公司。即使最後失敗,相信也會滋養出更大的輝煌。而一旦成功,便能奠定絕對的市場地位,讓其它競爭者難望其項背。

可以看到,學術界的成果和產品之間還有不小的鴻溝。當然其中的因素有很多,如成本、功耗等等,而其中最關鍵的因素之一是性能。傳統的方式很多時候會算法管算法,整好後拿去優化,相互獨立,最多整幾輪迭代。而今天我們看到,兩者需要越來越多地相互融合,共同演進。通過hardware-software co-design才能打造和打磨出更加完美的產品。它需要算法設計中便考慮對於特定平臺硬件上的友好性。舉例來說,爲了更好的部署,網絡設計時最好就要考慮哪些算子在目標平臺上能被較好地加速;同時訓練時加入特定的元素以便於後面的模型剪枝和量化。如果等吭哧吭哧訓練了幾周,模型都出來了再考慮這些問題就可能會帶來巨大的成本。近幾年大熱的AutoML中的自動神經網絡架構搜索(NAS)現在也越來越多地朝着hardware/platform-ware的方向發展。

最後,車輛環境感知中,數據的長尾問題是擺在AD/ADAS面前最大的問題。車輛環境是個開放環境,路上可能碰到任何無法預想的東西。2016年蘭德智庫指出自動駕駛系統需要進行110億英里的測試才能達到量產應用條件。顯然,這不是幾輛車上路滿大街跑能搞得定的,傳統的測試手段已捉襟見肘。當然,對於ADAS這類駕駛輔助類功能要求會低一些,但本質上面臨的問題是類似的。傳統的汽車功能安全標準已經無法涵蓋這類問題。雖然現在有針對性的預期功能安全(SOTIF)標準正在起草,但其可操作性和有效性還有待驗證。總得來說,汽車的智能化給測試驗證提出了非常有趣同時也是前所末有的挑戰。

參考資料

1] B. Huval et al., “An Empirical Evaluation of Deep Learning on Highway Driving,” CoRR, vol. abs/1504.01716, 2015.
[2] C. Szegedy, A. Toshev, and D. Erhan, “Deep Neural Networks for Object Detection,” in Advances in Neural Information Processing Systems 26, 2013, pp. 2553–2561.
[3] E. Shelhamer, J. Long, and T. Darrell, “Fully Convolutional Networks for Semantic Segmentation,” CoRR, vol. abs/1605.06211, 2016.
[4] V. Badrinarayanan, A. Handa, and R. Cipolla, “SegNet: A Deep Convolutional Encoder-Decoder Architecture for Robust Semantic Pixel-Wise Labelling,” CoRR, vol. abs/1505.07293, 2015.
[5] L.-C. Chen, G. Papandreou, I. Kokkinos, K. Murphy, and A. L. Yuille, “DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs,” CoRR, vol. abs/1606.00915, 2016.
[6] E. Romera, L. M. Bergasa, and R. Arroyo, “Can we unify monocular detectors for autonomous driving by using the pixel-wise semantic segmentation of CNNs?,” CoRR, vol. abs/1607.00971, 2016.
[7] J. Kim and C. Park, “End-To-End Ego Lane Estimation Based on Sequential Transfer Learning for Self-Driving Cars,” in 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), 2017, pp. 1194–1202.
[8] X. Pan, J. Shi, P. Luo, X. Wang, and X. Tang, “Spatial As Deep: Spatial CNN for Traffic Scene Understanding,” ArXiv E-Prints, Dec. 2017.
[9] Y.-C. Hsu, Z. Xu, Z. Kira, and J. Huang, “Learning to Cluster for Proposal-Free Instance Segmentation,” CoRR, vol. abs/1803.06459, 2018.
[10] S. Lee et al., “VPGNet: Vanishing Point Guided Network for Lane and Road Marking Detection and Recognition,” CoRR, vol. abs/1710.06288, 2017.
[11] D. Neven, B. De Brabandere, S. Georgoulis, M. Proesmans, and L. Van Gool, “Towards End-to-End Lane Detection: an Instance Segmentation Approach,” ArXiv E-Prints, Feb. 2018.
[12] B. D. Brabandere, W. V. Gansbeke, D. Neven, M. Proesmans, and L. V. Gool, “End-to-end Lane Detection through Differentiable Least-Squares Fitting,” CoRR, vol. abs/1902.00293, 2019.
[13] N. Garnett, R. Cohen, T. Pe’er, R. Lahav, and D. Levi, “3D-LaneNet: end-to-end 3D multiple lane detection,” CoRR, vol. abs/1811.10203, 2018.
[14] Y. Hou, Z. Ma, C. Liu, and C. Change Loy, “Learning Lightweight Lane Detection CNNs by Self Attention Distillation,” ArXiv E-Prints, p. arXiv:1908.00821, Aug. 2019.
[15] Z. Zou, Z. Shi, Y. Guo, and J. Ye, “Object Detection in 20 Years: A Survey,” CoRR, vol. abs/1905.05055, 2019.
[16] X. Wu, D. Sahoo, and S. C. H. Hoi, “Recent Advances in Deep Learning for Object Detection,” ArXiv E-Prints, p. arXiv:1908.03673, Aug. 2019.
[17] Q. Zhao, T. Sheng, Y. Wang, F. Ni, and L. Cai, “CFENet: An Accurate and Efficient Single-Shot Object Detector for Autonomous Driving,” ArXiv E-Prints, Jun. 2018.
[18] A. Bewley, Z. Ge, L. Ott, F. Ramos, and B. Upcroft, “Simple Online and Realtime Tracking,” CoRR, vol. abs/1602.00763, 2016.
[19] N. Wojke, A. Bewley, and D. Paulus, “Simple Online and Realtime Tracking with a Deep Association Metric,” CoRR, vol. abs/1703.07402, 2017.
[20] X. Pan, P. Luo, J. Shi, and X. Tang, “Two at Once: Enhancing Learning and Generalization Capacities via IBN-Net,” in The European Conference on Computer Vision (ECCV), 2018.
[21] H. Zhao et al., “PSANet: Point-wise Spatial Attention Network for Scene Parsing,” in Computer Vision – ECCV 2018, Cham, 2018, pp. 270–286.
[22] A. G. Howard et al., “MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications,” ArXiv E-Prints, Apr. 2017.
[23] M. Sandler, A. Howard, M. Zhu, A. Zhmoginov, and L.-C. Chen, “Inverted Residuals and Linear Bottlenecks: Mobile Networks for Classification, Detection and Segmentation,” ArXiv E-Prints, Jan. 2018.
[24] X. Zhang, X. Zhou, M. Lin, and J. Sun, “ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices,” ArXiv E-Prints, Jul. 2017.
[25] N. Ma, X. Zhang, H.-T. Zheng, and J. Sun, “ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design,” ArXiv E-Prints, 2018.
[26] M. Tan and Q. V. Le, “EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks,” ArXiv E-Prints, p. arXiv:1905.11946, May 2019.

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章