蕭簫 發(fā)自 凹非寺
量子位 報(bào)道 | 公眾號 QbitAI

見過3D物體數(shù)據(jù)集，見過會動的3D物體數(shù)據(jù)集嗎？

每段動態(tài)視頻都以目標(biāo)為中心拍攝，不僅自帶標(biāo)注整體的邊界框，每個(gè)視頻還附帶相機(jī)位姿和稀疏點(diǎn)云。

這是谷歌的開源3D物體數(shù)據(jù)集Objectron，包含15000份短視頻樣本，以及從五個(gè)大洲、十個(gè)國家里收集來的400多萬張帶注釋的圖像。

谷歌認(rèn)為，3D目標(biāo)理解領(lǐng)域，缺少像2D中的ImageNet這樣的大型數(shù)據(jù)集，而Objectron數(shù)據(jù)集能在一定程度上解決這個(gè)問題。

數(shù)據(jù)集一經(jīng)推出，1.6k網(wǎng)友點(diǎn)贊。

有網(wǎng)友調(diào)侃，谷歌恰好在自己想“谷歌”這類數(shù)據(jù)集的時(shí)候，把它發(fā)了出來。

也有團(tuán)隊(duì)前成員表示，很高興看到這樣的數(shù)據(jù)集和模型，給AR帶來進(jìn)步的可能。

除此之外，谷歌還公布了用Objectron數(shù)據(jù)集訓(xùn)練的針對鞋子、椅子、杯子和相機(jī)4種類別的3D目標(biāo)檢測模型。

來看看這個(gè)數(shù)據(jù)集包含什么，以及谷歌提供的3D目標(biāo)檢測方案吧~(項(xiàng)目地址見文末)

9類物體，對AR挺友好

目前，這個(gè)數(shù)據(jù)集中包含的3D物體樣本，包括自行車，書籍，瓶子，照相機(jī)，麥片盒子，椅子，杯子，筆記本電腦和鞋子。

當(dāng)然，這個(gè)數(shù)據(jù)集，絕不僅僅只是一些以物體為中心拍攝的視頻和圖像，它具有如下特性：

注釋標(biāo)簽(3D目標(biāo)立體邊界框)

用于AR數(shù)據(jù)的數(shù)據(jù)(相機(jī)位姿、稀疏點(diǎn)云、二維表面)

數(shù)據(jù)預(yù)處理(圖像格式為tf.example，視頻格式為SequenceExample)

支持通過腳本運(yùn)行3D IoU指標(biāo)的評估

支持通過腳本實(shí)現(xiàn)Tensorflow、PyTorch、JAX的數(shù)據(jù)加載及可視化，包含“Hello World”樣例

支持Apache Beam，用于處理谷歌云(Google Cloud)基礎(chǔ)架構(gòu)上的數(shù)據(jù)集

所有可用樣本的索引，包括訓(xùn)練/測試部分，便于下載

圖像部分的畫風(fēng)，基本是這樣的，也標(biāo)注得非常詳細(xì)：

而在視頻中，不僅有從各個(gè)角度拍攝的、以目標(biāo)為中心的片段(從左到右、從下到上)：

也有不同數(shù)量的視頻類型(一個(gè)目標(biāo)、或者兩個(gè)以上的目標(biāo))：

谷歌希望通過發(fā)布這個(gè)數(shù)據(jù)集，讓研究界能夠進(jìn)一步突破3D目標(biāo)理解領(lǐng)域，以及相關(guān)的如無監(jiān)督學(xué)習(xí)等方向的研究應(yīng)用。

怎么用？谷歌“以身示范”

拿到數(shù)據(jù)集的第一刻，并不知道它是否好用，而且總感覺有點(diǎn)無從下手？

別擔(dān)心，這個(gè)數(shù)據(jù)集的訓(xùn)練效果，谷歌已經(jīng)替我們試過了。

看起來還不錯(cuò)：

此外，谷歌將訓(xùn)練好的3D目標(biāo)檢測模型，也一并給了出來。(傳送見文末)

算法主要包括兩部分，第一部分是Tensorflow的2D目標(biāo)檢測模型，用來“發(fā)現(xiàn)物體的位置”；

第二部分則進(jìn)行圖像裁剪，來估計(jì)3D物體的邊界框(同時(shí)計(jì)算目標(biāo)下一幀的2D裁剪，因此不需要運(yùn)行每個(gè)幀)，整體結(jié)構(gòu)如下圖：

在模型的評估上，谷歌采用了Sutherland-Hodgman多邊形裁剪算法，來計(jì)算兩個(gè)立體邊界框的交點(diǎn)，并計(jì)算出兩個(gè)立方體的相交體積，最終計(jì)算出3D目標(biāo)檢測模型的IoU。

簡單來說，兩個(gè)立方體重疊體積越大，3D目標(biāo)檢測模型效果就越好。

這個(gè)模型是谷歌推出的MediaPipe中的一個(gè)部分，后者是一個(gè)開源的跨平臺框架，用于構(gòu)建pipeline，以處理不同形式的感知數(shù)據(jù)。

它推出的MediaPipe Objectron實(shí)時(shí)3D目標(biāo)檢測模型，用移動設(shè)備(手機(jī))就能進(jìn)行目標(biāo)實(shí)時(shí)檢測。

看，(他們玩得多歡快)實(shí)時(shí)目標(biāo)檢測的效果還不錯(cuò)：

其他部分3D數(shù)據(jù)集

除了谷歌推出的數(shù)據(jù)集以外，此前視覺3D目標(biāo)領(lǐng)域，也有許多類型不同的數(shù)據(jù)集，每個(gè)數(shù)據(jù)集都有自己的特點(diǎn)。

例如斯坦福大學(xué)等提出的ScanNetV2，是個(gè)室內(nèi)場景數(shù)據(jù)集，而ScanNet則是個(gè)RGB-D視頻數(shù)據(jù)集，一共有21個(gè)目標(biāo)類，一共1513個(gè)采集場景數(shù)據(jù)，可做語義分割和目標(biāo)檢測任務(wù)。

而目前在自動駕駛領(lǐng)域非常熱門的KITTI數(shù)據(jù)集，也是一個(gè)3D數(shù)據(jù)集，是目前最大的自動駕駛場景下計(jì)算機(jī)視覺的算法評測數(shù)據(jù)集，包含市區(qū)、鄉(xiāng)村和高速公路等場景采集的真實(shí)圖像數(shù)據(jù)。

此外，還有Waymo、SemanticKITTI、H3D等等數(shù)據(jù)集，也都用在不同的場景中。(例如SemanticKITTI，通常被專門用于自動駕駛的3D語義分割)

無論是視頻還是圖像，這些數(shù)據(jù)集的單個(gè)樣本基本包含多個(gè)目標(biāo)，使用場景上也與谷歌的Objectron有所不同。

感興趣的小伙伴們，可以通過下方傳送門，瀏覽谷歌最新的3D目標(biāo)檢測數(shù)據(jù)集，以及相關(guān)模型~

Objectron數(shù)據(jù)集傳送門：https://github.com/google-research-datasets/Objectron/

針對4種物體的3D目標(biāo)檢測模型：https://google.github.io/mediapipe/solutions/objectron

參考鏈接： https://ai.googleblog.com/2020/11/announcing-objectron-dataset.htmlhttps://ai.googleblog.com/2020/03/real-time-3d-object-detection-on-mobile.html

— 完 —

量子位 QbitAI · 頭條號簽約

關(guān)注我們，第一時(shí)間獲知前沿科技動態(tài)

總結(jié)

以上是生活随笔為你收集整理的疲劳驾驶样本集_谷歌AI最新3D数据集，1.5万张动图，让AR主宰你的生活的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。