最新的一篇MOT領域基于Transformer的工作，是第一個真正嚴格意義上端到端的多目標跟蹤框架，超越了此前的TransTrack和TrackFormer等工作。

簡介

多目標跟蹤的關鍵挑戰(zhàn)在于軌跡上目標的時序建模，而現(xiàn)有的TBD方法大多采用簡單的啟發(fā)式策略，如空間和外觀相似度。盡管這些方法具有通用性，但它們過于簡單，不足以對復雜的變化進行建模，例如通過遮擋進行跟蹤。 本質(zhì)上，現(xiàn)有方法缺乏時間建模的能力。 這篇論文中，作者提出了MOTR，這是一個真正的完全端到端的跟蹤框架。MOTR能夠?qū)W習建模目標的長程時間變化，它隱式地進行時間關聯(lián)，并避免了以前的顯式啟發(fā)式策略?；赥ransformer和DETR，MOTR引入了track query這個概念，一個track query負責建模一個目標的整個軌跡，它可以在幀間傳輸并更新從而無縫完成目標檢測和跟蹤任務。時間聚合網(wǎng)絡（temporal aggregation network，TAN）配合多幀訓練被用來建模長程時間關系。實驗結果表明MOTR達到了SOTA效果。

• 論文標題

  MOTR: End-to-End Multiple-Object Tracking with TRansformer

• 論文地址

  http://arxiv.org/abs/2105.03247

• 論文源碼

  https://github.com/megvii-model/MOTR

介紹

多目標跟蹤是在視頻序列的每一幀中定位所有的目標并確定它們的移動軌跡的一個任務。多目標跟蹤是極具挑戰(zhàn)的一個任務，因為每幀中的目標都可能因為環(huán)境的變化而被遮擋，而且跟蹤器要想進行長期跟蹤或者低幀率的跟蹤是比較困難的。這些復雜多樣的跟蹤場景為MOT方法的設計帶來了諸多挑戰(zhàn)。

現(xiàn)有的多目標跟蹤方法基本上都遵循tracking-by-detection（TBD）范式，它將軌跡的生成分為兩個步驟：目標定位和時序關聯(lián)。對目標定位而言，使用檢測器逐幀檢測目標即可。而對于時序關聯(lián)，現(xiàn)有的方法要么使用空間相似性（即基于IoU關聯(lián)）要么使用外觀相似性（即基于ReID關聯(lián)）。對于基于IoU的方法，計算兩幀檢測框的兩兩之間的IoU矩陣，若兩個目標之間的IoU高于某個閾值則賦予同一個ID，如下圖的(a)所示?；赗eID的方法思路類似，兩幀之間目標兩兩計算特征相似度，具有最高相似度的兩個目標賦予同一個ID，不過，單獨訓練一個檢測器和ReID模型代價太大，最近的主流思路是一個模型聯(lián)合訓練檢測和ReID分支，這類方法稱為JDT（joint detection and tracking）方法，如下圖(b)所示。

上述的時序關聯(lián)方法都是啟發(fā)式的，并且是相當簡單的，因此它們難以建模長時間的物體復雜的空間和外觀變化。本質(zhì)上看，其不具備對時間變化建模的能力，這和深度學習“端到端學習”的理念是不一致的。這些方法也許大多數(shù)情況下工作正常，但是對于一些挑戰(zhàn)性的場景缺乏魯棒性，這些場景下IoU和外觀是不可信的，而這些復雜場景才是MOT任務的關鍵。因此，作者為了解決這個問題，構建了一個不需要任何數(shù)據(jù)關聯(lián)處理的端到端跟蹤框架。

最近，DETR提出了一套端到端目標檢測的策略，它開創(chuàng)性地提出了“object query”這個概念，這是目標的顯式解耦表示（representation），這種表示簡化了Transformer框架的學習。首次啟發(fā)，論文作者拓展“object query”到目標跟蹤領域，形成了“track query”，每個track query負責預測一個對象的完整軌跡。如上圖?所示，和分類和回歸分支并行，MOTR對每一幀預測track query集合，這個track query集輸入到decoder網(wǎng)絡中以產(chǎn)生當前幀的跟蹤預測和更新的track query，更新后的track query傳遞到下一幀的decoder中。這個query傳遞和預測的處理過程會在整個視頻序列上一幀幀重復進行，因此稱為連續(xù)query傳遞（continuous query passing）。由于一個query一旦和某個目標匹配，這個query就會一直跟隨這個目標，因此continuous query passing可以很自然地消除數(shù)據(jù)關聯(lián)和一些手工操作，如NMS。

接著，為了建模長期時間關系，作者進一步引入了多幀訓練以及時間聚合網(wǎng)絡（TAN），TAN建立了一個query memory bank來收集歷史幀中和已跟蹤目標對應的query，當前幀的track query會和memory 板塊中的每個query通過多頭注意力進行交互。

作者認為，MOTR是簡單且高效的，并且嚴格意義上它是第一個端到端的MOT框架。通過多幀時間聚合學習的track query有強大的時間建模能力，因而MOTR不需要時間關聯(lián)處理或者手工操作。相反，此前基于Transformer的工作并非完全端到端的，它們依然需要IoU匹配或者Tracking NMS等操作。在MOT16和MOT17上的實驗結果表明，MOTR達到SOTA性能并且比之前的方法具有更低的復雜度。

MOTR

Deformable DETR

DETR是Transformer在目標檢測領域的經(jīng)典成功案例之一。在DETR中，object query是固定數(shù)量的可學習位置嵌入，表示可能的目標的proposal。一個object query只會對應一個目標，這種對應關系是通過二分圖匹配構建的。考慮到DETR較高的復雜度和較慢的收斂速度，Deformable DETR將Transformer中的self-attention替換為了multi-scale deformable attention。為了表示object query如何通過decoder與特征交互，作者重新設計了Deformable DETR的解碼器。

令$q\in R^C$表示object query集且$f\in R^C$表示從encoder獲得的特征圖，這里的$C$表示特征的維度。解碼器的處理過程可以描述如下式，其中$k\in 1,\dots ,K$（$K$為decoder的層數(shù)），$q^k$表示第$k$個decoder層的output query。$G_{sa}$表示DETR中的self-attention操作，$G_{ca}$則表示multi-scale deformable attention操作。

$$q^k=G_{ca}\left(G_{sa}\left(q^{k?1}\right),f\right)$$

Framework

MOTR中，作者引入track query和continuous query passing來進行端到端的跟蹤，temporal aggregation network進一步被提出用來增強多幀的時間信息。下面就逐一闡述這幾個核心模塊。

End-to-End Track Query Passing

Track Query:

DETR中的object query并不負責預測特定的目標，因此一個object query在不同的幀上可能預測不同的目標。如上圖所示，DETR對MOT數(shù)據(jù)集進行檢測，同一個object query在兩個不同幀中預測的目標是不一樣的（如上圖中(a)綠色框所示），因此將query的id用于軌跡關聯(lián)是不合理的。（作者這里其實想表述的是，object query只是對一個區(qū)域目標負責，并不具體到id級別。）

針對上述問題，作者拓展object query進行跟蹤任務，拓展后的稱為track query，每個track query負責一個目標的整個軌跡的預測。如上圖(b)所示，一旦一個track query在某一幀和某個目標匹配之后，這個track query將一直預測這個目標直到該目標消失。因此，同一個track query的所有預測結果就形成了一個目標的軌跡，不需要顯式的數(shù)據(jù)關聯(lián)步驟，至于track query之間的order-preserving則是依據(jù)特定目標的track query監(jiān)督的。當然，還有一個問題，那就是新目標的產(chǎn)生，因此作者進一步提出了empty query來負責新目標的檢測。

Continuous Query Passing:

基于上述提出的track query，作者進一步提出了continuous query passing機制，在這個機制下，track query隨著幀的變化而改變representation和匹配的目標的localization。然后，作者提出了基于continuous query passing的端到端MOT框架，即MOTR。在MOTR中，目標的時間變化建模是編碼器的多頭注意力隱式學習的，因此不需要顯式的數(shù)據(jù)關聯(lián)。

MOTR的整體結構如上圖所示，視頻序列首先會送入CNN中，隨后進入Deformable DETR的編碼器提取基本特征$f=\left\{f_0,f_1,\dots ,f_N\right\}$，這里的$f_0$表示第$T_0$幀的特征。對第$T_0$幀而言，特征$f_0$和empty query set $q_e$被輸入到decoder網(wǎng)絡中定位所有初始化目標并生成原始的track query set $q_{ot}^1$。到這里，對應到上圖就是最左側的那部分，需要注意，這里的continuous query passing其實將$T_0$的$q_{ot}^0$傳遞到了下一幀。 $q_{ot}^0$通過QIM模塊生成了下一幀（
$T_1$）的track query輸入，也就是圖上對應的$q_t^1$。所以從上圖可以看出來，整個模型迭代式地處理每一幀$T_i$（$i\in [1,N]$），QIM根據(jù)上一幀地輸出產(chǎn)生的$q_t^i$會和empty query set $q_e$級聯(lián)到一起，級聯(lián)后的query set $q_e$回合特征$f_i$一起送入decoder中直接產(chǎn)生當前幀的預測結果，并且更新query set $q_{ot}^{i+1}$將其送入到下一幀。

Query Interaction Module

在上面的敘述中，QIM負責接受上一幀的track query輸出并生成當前幀的track query輸入，在這一節(jié)將具體闡述Query Interaction Module（QIM）。QIM主要包括目標進出機制（object entrance and exit mechanism）和時間聚合網(wǎng)絡（temporal aggregation network）。

Object Entrance and Exit: 首先來看目標進出機制，我們知道，每個track query表示一個完整軌跡，然而，一些目標可能在中間某一幀出現(xiàn)或者消失，因此MOTR需要輸出一些邊界框$\left\{bo{x}_i,\dots ,bo{x}_j\right\}$假定目標在$T_i$幀出現(xiàn)但在$T_j$幀消失。

MOTR是如何處理目標進出的情況呢？在訓練時，track query的學習可以通過二分圖匹配的GT來監(jiān)督。但是，在推理時，使用跟蹤得分預測來決定一個軌跡的出現(xiàn)和消失。來看下圖，這是QIM的結構圖，對$T_i$幀而言，track query set $q_t^i$通過QIM從$T_{i?1}$幀生成，然后和empty query set $q_e$級聯(lián)到一起，級聯(lián)的結果繼而輸入到decoder并產(chǎn)生原始的包含跟蹤得分的track query set $q_{ot}^i$。$q_{ot}^i$隨機被分割為兩個query set，即$q_{en}^i=q_{\text{ot?}}^i\left[:d_e\right]$和$q_{ce}^i=q_{ot}^i\left[d_e:\right]$，這里的$d_e$是$q_e$中query的數(shù)量。$q_{en}^i$包含進入的目標而$q_{ce}^i$包含跟蹤著的和離開的目標。對目標的進入，$q_{en}^i$中的query（下圖的“3”）如果跟蹤得分大于進入閾值${\tau }_{en}$則被保留，其余的被移除。表示如下式，$s_k$表示$q_{en}^i$的第$k$個query $q_k$的分類得分。

$${\bar{q}}_{en}^i=\left\{q_k\in q_{en}^i\mid s_k>{\tau }_{en}\right\}$$

對目標的退出，$q_{ce}^i$的query（上圖的“2”）如果跟蹤得分連續(xù)M幀低于退出閾值${\tau }_{ex}$，將被移除，剩下的query（上圖的“1”）則被保留，式子如下，實驗中設置${\tau }_{en}=0.8,{\tau }_{ex}=0.6$且$M=5$。

$${\bar{q}}_c^i=\left\{q_k\in q_{ce}^i\mid \max \left\{s_k^i,\dots ,s_k^{i?M}\right\}>{\tau }_{ex}\right\}$$

Temporal Aggregation Network: 接著，來看時間聚合網(wǎng)絡TAN，TAN的目的是增強時間相關性并為跟蹤目標提供上下文先驗信息。還是上面那張圖，首先構建了一個memory bank以便于跟蹤目標的時間聚合。memory bank $q_{bank}=\left\{q_c^{i?M},\dots ,{\tilde{q}}_c^i\right\}$首先收集歷史query（也就是和已跟蹤目標相關聯(lián)的query），然后這些memory bank中的query將被級聯(lián)到一起，如下式所示。

$$tgt=q_c^{i?M}\oplus ?{\tilde{q}}_c^{i?1}\oplus {\tilde{q}}_c^i$$

級聯(lián)之后，這些query被送入多頭注意力模塊，它們既是value也是key，繼而生成注意力權重。${\bar{q}}_c^{?}$被作為MHA（多頭注意力）的query，因此有下面的點積注意力的式子，其中的${\sigma }_s$表示softmax函數(shù)而$d$表示track query的維度。

$$q_{sa}^i={\sigma }_s\left(\frac{tgt?\mathrm{tg}{t}^T}{\sqrt{d}}\right)?{\bar{q}}_c^i$$

之后，$q_{sa}^i$被FFN網(wǎng)絡進一步調(diào)整，這里的FC表示線性投影層而LN表示層標準化，${\sigma }_r$表示ReLU激活函數(shù)。

$$\begin{array}{c}t\tilde{g}t=LN\left(q_{sa}^i+{\bar{q}}_c^i\right)\\ {\hat{q}}_c^i=LN(FC\left({\sigma }_r(FC(\widetilde{gg}t))+t\tilde{g}t\right)\end{array}$$

TAN的輸出${\hat{q}}_c$和${\bar{q}}_{en}^i$串聯(lián)到一起來產(chǎn)生$T_{i+1}$幀的track query set $q_t^{i+1}$。

Overall Optimization

既然是端到端的結構，那么優(yōu)化就顯得非常重要，MOTR是以視頻序列作為輸入并且逐幀計算跟蹤損失的，整體的跟蹤損失是每一幀的損失的和并根據(jù)GT數(shù)目標準化的結果，如下式所示，這里的$N$表示視頻序列的長度，$Y_i$和${\hat{Y}}_i$分別表示$T_i$幀的預測和對應的GT。$V_i=V_t^i+V_e^i$表示$T_i$幀上所有的GT數(shù)目，$V_t^i$和$V_e^i$分別是已跟蹤的目標數(shù)目和新軌跡的數(shù)目。

$$L_{ot}(Y,\hat{Y})=\frac{{\sum }_{n=0}^N\left(L_t\left(Y_i,{\hat{Y}}_i\right)\right)}{{\sum }_{n=0}^N\left(V_i\right)}$$

$L_t$是單幀的跟蹤損失，和Deformable DETR中的檢測損失是類似的，單幀損失$L_t$可以用下面的式子表示，其中$L_{cls}$是focal loss，$L_{l_1}$表示L1損失而$L_{\text{giou?}}$表示GIoU損失，${\lambda }_{cls},{\lambda }_{l_1}$和${\lambda }_{\text{giou?}}$則是相應的權重系數(shù)。

$$L_t\left(Y_i,{\hat{Y}}_i\right)={\lambda }_{cls}{L}_{cls}+{\lambda }_{l_1}{L}_{l_1}+{\lambda }_{\text{giou?}}{L}_{\text{giou?}}$$

MOTR的跟蹤損失和Deformable DETR的檢測損失主要的不同在于標簽分配的方式不同，對檢測損失而言，標簽的分配完全由匈牙利的分配結果決定，但是對跟蹤損失而言，每個track query負責特定的目標，因此其GT是由對應ID的目標決定的，對empty query而言，它的GT才是通過匈牙利匹配得到的。

實驗

首先是和其他SOTA的比較，MOTR確實取得了相當不錯的效果，相比此前基于Transformer的方法也有了不小的提高。

也進行了一些模塊的消融實驗，如下。

總結

論文基于Transfomer提出了一個真正的端到端的多目標跟蹤框架MOTR并超越了之前的方法，該框架真正意義上不需要數(shù)據(jù)關聯(lián)和NMS等后處理操作，并且在基準數(shù)據(jù)集上達到了SOTA效果，是很值得關注的工作。本文也只是我本人從自身出發(fā)對這篇文章進行的解讀，想要更詳細理解的強烈推薦閱讀原論文。最后，如果我的文章對你有所幫助，歡迎一鍵三連，你的支持是我不懈創(chuàng)作的動力。

超強干貨來襲 云風專訪：近40年碼齡，通宵達旦的技術人生

總結

以上是生活随笔為你收集整理的MOTR解读的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

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