Siamese RPN

Paper : High Performance Visual Tracking with Siamese Region Proposal Network
Code : unofficial

摘要

之前在阅读SiamFC的时候就感觉目标追踪与物体检测在候选框生成上可以有很多互相借鉴的地方,而且SiamFC存在两个问题:一个是遍历的方法容易成为时间瓶颈,另一个是定位框处理有点太粗糙了,不仅固定了尺寸还固定了长宽比。在目标检测上提出的RPN迟早要迁移到物体追踪的任务中。因此商汤在CVPR2018中掏出了一篇SiamRPN。对于深度较深的CNN处理目标追踪,基本上为了保证算法的实时性很难像相关滤波器一样进行在线训练,因为更新参数时间花费太高,只能离线训练。基本上都是one-shot detection task,也就是把每一帧和第一帧中目标的图像拿出来进行比较。

相比于基于滤波器的目标追踪算法,作者认为SiamRPN作为没有在线finetune的算法,表现依旧比较好的原因有以下两点

  1. 足够大的训练数据
  2. RPN预测的比较准

网络架构

在这里插入图片描述
前半部分属于传统的SiamNetwork,两CNN共享参数进行特征提取,中间是一个RPN网络,一支用来进行二分类,而另一支用来在archors的基础上对定位框进行进一步的回归,其中k表示RPN中选定的archor个数。经过CNN模块后,将RPN每个分支中的两个输出进行相关操作,使用 ⋆ ⋆ 表示相关操作,得到输出

A w × h × 2 k c l s = [ φ ( x ) ] c l s ⋆ [ φ ( z ) ] c l s A w × h × 4 k r e g = [ φ ( x ) ] r e g ⋆ [ φ ( z ) ] r e g \\A_{w\times h\times 2k}^{cls} = [\varphi(x)]_{cls}⋆ [\varphi(z)]_{cls} \\A_{w\times h\times 4k}^{reg} = [\varphi(x)]_{reg}⋆ [\varphi(z)]_{reg} Aw×h×2kcls=[φ(x)]cls[φ(z)]clsAw×h×4kreg=[φ(x)]reg[φ(z)]reg

训练时使用的损失函数与Faster RCNN 中的相同,对cls使用交叉熵,对reg使用smooth-L1。因为相邻两帧之间尺寸变化不大,因此选择archor时只使用一种尺寸,多种长宽比 [ 0.33 , 0.5 , 1 , 2 , 3 ] [0.33,0.5,1,2,3] [0.33,0.5,1,2,3] 。训练时,template 和 detection patch pair从同一个视频中抽取两帧生成。SiamRPN的每个分支都在ImageNet数据集上进行了预训练。正负训练样本定义如下:定义Archor与目标定位框IoU>0.6 为正例,<0.3为负例。

one-shot 部分的解释对应下图

在这里插入图片描述
图中,将Template frame 计算一次之后,把上半部分的输出当成下半部分网络的参数,然后将上半部分裁剪掉,该网络就对应了一个单输入的目标追踪网络了。

最后,是proposal region 的选取问题,有两种策略进行选取

  • 舍弃 archor 距离图片中心较远的proposal
    在这里插入图片描述

  • 使用cosine窗口对不同位置的score进行加权,根据候选框的长宽变化对proposal进行惩罚

    penalty = exp ⁡ ( k max ⁡ ( r r ′ , r ′ r ) max ⁡ ( s s ′ , s ′ s ) ) \text{penalty}= \exp(k\max(\frac{r}{r'},\frac{r'}{r})\max(\frac{s}{s'},\frac{s'}{s})) penalty=exp(kmax(rr,rr)max(ss,ss))
    其中, k k k 是超参数, r , r ′ r,r' r,r 是当前帧与上一帧的长宽比, s , s ′ s,s' s,s是当前帧与上一帧的尺寸
    ( w + p ) ( h + p ) = s 2 (w+p)(h+p) = s^2 (w+p)(h+p)=s2
    p p p 表示padding宽度, p = w + h 2 p=\frac{w+h}{2} p=2w+h 。最后,在对分数进行重计算和NMS之后,得到当前帧的定位框位置

总结

SiamRPN在所有实时算法里准确率最高,完全达到了实时tracking的SOTA,SiamRPN称得上应用性很强的算法。但是Siam RPN在结构上没有改进,因此没能拿到oral。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值