視覺目標跟蹤SiamRPN

Paper:High Performance Visual Tracking with Siamese Region Proposal Network兄纺。
http://openaccess.thecvf.com/content_cvpr_2018/papers/Li_High_Performance_Visual_CVPR_2018_paper.pdf

我們在前面已經介紹了SiamFC,SiamFC的方法只能估計目標的中心位置,而要想對目標的尺寸進行估計,只有通過多尺度測試來預測尺度的變化,這種方式不僅增加了計算量,同時也不夠精確。SiamRPN作為對其的改進篡殷,主要通過引入?yún)^(qū)域候選網絡(Region Proposal Networks, RPN)來對bbox進行回歸,提高跟蹤定位的精度埋涧。

1. 算法原理

1.1 SiamRPN網絡架構

由上圖可以看出板辽,整個網絡結構可以分為兩部分:

  • Siames Network,與SiamFC類似棘催,孿生網絡分為上下兩支劲弦,上下兩支路的網絡結構和參數(shù)完全相同,該網絡的作用是分別提取模板幀和檢測幀的圖像特征醇坝。孿生網絡的兩個分支可以用一個卷積網絡實現(xiàn)邑跪,值得注意的是這個卷積網絡必須為全卷積網絡,滿足平移不變性呼猪,即先對圖像進行有比例因子的轉換操作再進行全卷積操作等同于先對圖像進行全卷積操作再進行轉換操作画畅。
  • Region Proposal Network,該子網絡的作用是對bbox進行回歸宋距,得到精確的位置估計轴踱。RPN網絡由兩部分組成,一部分是分類分支谚赎,用于區(qū)分目標和背景淫僻,另一部分是回歸分支,它將候選區(qū)域進行微調壶唤。

1.2 Anchors

在討論RPN前雳灵,需要說一說anchor,anchor的本質就是輸出一系列不同尺寸和形狀的矩形闸盔,

在RPN中使用滑窗技術在特征圖的每個點都生成k個anchor细办,即通過這些矩形來對bbox進行回歸。

1.3 RPN

從網絡架構可以看出蕾殴,模板圖像和搜索圖像經過Siamese網絡分別得到6 x 6 x 256,22 x 22 x 256的特征岛啸,然后模板圖像特征通過3 x 3的卷積核分別產生了4 x 4 x (2k x 256)以及4 x 4 x (4k x 256)的特征钓觉,這里從6 x 6的尺寸經過3 x 3的卷積核得到4 x 4的特征尺寸比較簡單,值得注意的是特征通道從256上升到了2k x 256以及4k x 256坚踩,之所以通道數(shù)上升了2k倍荡灾,是在特征圖的每個點生成k個anchor,同時每個anchor可以被分類到前景或背景,所以分類分支上升了2k倍批幌,同理每個anchor可以用4個參數(shù)進行描述础锐,所以回歸分支上升了4k倍。同時搜索圖像也通過3 x 3的卷積核分別得到兩個特征荧缘,這里特征通道數(shù)保持不變皆警。

對于分類分支,將2k個模板圖像anchor的4 x 4 x 256特征作為卷積核與搜索圖像特征進行卷積操作截粗,從而產生分類分支響應圖信姓;對于回歸分支與此類似,產生的響應圖為17 x17 x 4k绸罗,其中每個點表示一個尺寸為4k的向量意推,為dx,dy珊蟀,dw菊值,dh,衡量著anchor與groundtruth的偏差育灸。響應圖計算公式如下:

1.4 訓練階段

在訓練過程中使用Faster R-CNN的loss函數(shù)腻窒,用cross-entropy loss作為分類分支的損失函數(shù),用smooth L1 loss作為回歸分支的損失函數(shù)描扯。

交叉熵loss與SiamFC類似定页。對于回歸分支的損失函數(shù),首先將anchor的坐標標準化绽诚,

smooth L1 loss函數(shù)為典徊,
總的loss函數(shù)為,
其中回歸loss為恩够,
Smooth L1
  • 相比于L1損失函數(shù)卒落,可以收斂得更快。
  • 相比于L2損失函數(shù)蜂桶,對離群點儡毕、異常值不敏感,梯度變化相對更小扑媚,訓練時不容易跑飛腰湾。

訓練過程中anchor數(shù)量一共五個,one scale, 5 ratios([0.33, 0.5, 1, 2, 3])疆股。

正負樣本選擇策略:設置高低兩個閾值费坊,大于0.6的為正樣本,小于0.3的為負樣本旬痹,一個pair中有64個樣本附井,其中正樣本最多16個讨越。

1.5 跟蹤階段

作者將跟蹤任務規(guī)劃為一個one-shot detection任務,即首先學習一個learner net永毅,在本文中對應于相似性函數(shù)把跨。學習完成后,在跟蹤階段通過初始幀學習得到卷積操作的卷積核參數(shù)沼死,從而得到pupil net着逐,然后再后續(xù)幀對目標進行跟蹤。其實在之前的相關濾波器類跟蹤器漫雕,SiamFC跟蹤器都存在這樣的思想滨嘱,SiamRPN從理論的角度對其進行了闡述。

跟蹤框架可以描述為浸间,

推理階段:

定義分類特征圖和回歸特征圖分別為太雨,

這里l為奇數(shù)對應于正樣本,選擇分類響應圖中top K個正樣本魁蒜,得出對應的anchors囊扳,然后得到與之對應的回歸響應圖中K個坐標偏移,最后通過anchors和這些偏移值得到預測值兜看,計算公式如下:

由此我們得到K個proposals锥咸。

Proposal selection:
我們需要從K個proposal中進行選取,文章中給出了兩個策略:

  • 作者認為相鄰幀目標的位移不會太大细移,所以只選取分類特征圖中心區(qū)域的點搏予。
  • 使用cosine窗和尺度變化懲罰來re-rank proposals的得分

最終使用NMS 來得到最終的bbox。

2 實驗

實驗環(huán)境是Intel i7弧轧、12G RAM雪侥、Nvidia GTX1060。

上圖為在OTB2015上的實驗結果精绎,可以看出相對于SiamFC跟蹤性能有利很大的提升速缨。同時跟蹤速度達到了160FPS,幾乎為SiamFC(86FPS)跟蹤速度的2倍代乃。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末旬牲,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子搁吓,更是在濱河造成了極大的恐慌原茅,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,651評論 6 501
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件堕仔,死亡現(xiàn)場離奇詭異擂橘,居然都是意外死亡,警方通過查閱死者的電腦和手機贮预,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,468評論 3 392
  • 文/潘曉璐 我一進店門贝室,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人仿吞,你說我怎么就攤上這事滑频。” “怎么了唤冈?”我有些...
    開封第一講書人閱讀 162,931評論 0 353
  • 文/不壞的土叔 我叫張陵峡迷,是天一觀的道長。 經常有香客問我你虹,道長绘搞,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,218評論 1 292
  • 正文 為了忘掉前任傅物,我火速辦了婚禮夯辖,結果婚禮上,老公的妹妹穿的比我還像新娘董饰。我一直安慰自己蒿褂,他們只是感情好,可當我...
    茶點故事閱讀 67,234評論 6 388
  • 文/花漫 我一把揭開白布卒暂。 她就那樣靜靜地躺著啄栓,像睡著了一般。 火紅的嫁衣襯著肌膚如雪也祠。 梳的紋絲不亂的頭發(fā)上昙楚,一...
    開封第一講書人閱讀 51,198評論 1 299
  • 那天,我揣著相機與錄音诈嘿,去河邊找鬼堪旧。 笑死,一個胖子當著我的面吹牛永淌,可吹牛的內容都是我干的崎场。 我是一名探鬼主播,決...
    沈念sama閱讀 40,084評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼遂蛀,長吁一口氣:“原來是場噩夢啊……” “哼谭跨!你這毒婦竟也來了?” 一聲冷哼從身側響起李滴,我...
    開封第一講書人閱讀 38,926評論 0 274
  • 序言:老撾萬榮一對情侶失蹤螃宙,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后所坯,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體谆扎,經...
    沈念sama閱讀 45,341評論 1 311
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,563評論 2 333
  • 正文 我和宋清朗相戀三年芹助,在試婚紗的時候發(fā)現(xiàn)自己被綠了堂湖。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片闲先。...
    茶點故事閱讀 39,731評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖无蜂,靈堂內的尸體忽然破棺而出伺糠,到底是詐尸還是另有隱情,我是刑警寧澤斥季,帶...
    沈念sama閱讀 35,430評論 5 343
  • 正文 年R本政府宣布训桶,位于F島的核電站,受9級特大地震影響酣倾,放射性物質發(fā)生泄漏舵揭。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,036評論 3 326
  • 文/蒙蒙 一躁锡、第九天 我趴在偏房一處隱蔽的房頂上張望午绳。 院中可真熱鬧,春花似錦稚铣、人聲如沸箱叁。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,676評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽耕漱。三九已至,卻和暖如春抬伺,著一層夾襖步出監(jiān)牢的瞬間螟够,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 32,829評論 1 269
  • 我被黑心中介騙來泰國打工峡钓, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留妓笙,地道東北人。 一個月前我還...
    沈念sama閱讀 47,743評論 2 368
  • 正文 我出身青樓能岩,卻偏偏與公主長得像寞宫,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子拉鹃,可洞房花燭夜當晚...
    茶點故事閱讀 44,629評論 2 354