光学学报, 2020, 40 (9): 0915003, 网络出版: 2020-05-06   

基于目标感知特征筛选的孪生网络跟踪算法 下载: 1469次

Tracking Algorithm for Siamese Network Based on Target-Aware Feature Selection
作者单位
1 燕山大学工业计算机控制工程河北省重点实验室, 河北 秦皇岛 066004
2 燕山大学国家冷轧板带装备及工艺工程技术研究中心, 河北 秦皇岛 066004
3 国网黑龙江省电力有限公司佳木斯供电公司, 黑龙江 佳木斯 154002
4 燕山大学电气工程学院, 河北 秦皇岛 066004
摘要
孪生网络跟踪算法是利用离线训练好的网络提取目标特征并进行匹配,从而实现跟踪。而离线训练深度特征在表征任意形式目标时将目标从背景中分离开的性能较差。为此,提出一种基于目标感知特征筛选的孪生网络跟踪算法。将经过裁剪处理后的模板帧和检测帧送入到ResNet50的特征提取网络分别提取目标和搜索区域的浅层、中层、深层特征;在目标感知模块中,通过设计一个回归损失函数来学习对目标敏感的特征,根据反向传播的梯度来确定每个卷积核的重要性程度,并以此来激活相对重要的卷积核筛选较重要的目标感知特征;将筛选得到的特征送入到SiamRPN模块,进行目标、背景的二分类判别和边界框的坐标回归,从而得到一个精确的目标边界框。在OTB2015和VOT2018两个标准数据集上进行测试实验,结果表明该算法可以实现对目标的稳健性跟踪。
Abstract
Tracking algorithms implemented in Siamese networks utilize an offline training network to extract features from a target object for matching and tracking. The offline-trained deep features are less efficient for distinguishing targets with arbitrary forms from the background. Therefore, we proposed a tracking algorithm for a Siamese network based on target-aware feature selection. First, the cropped template and detection frames were sent to a feature extraction network based on ResNet50 to extract the shallow, middle and deep features of the target and search regions. Second, in the target-aware module, a regression loss function was formulated for target-aware features and an importance scale for each convolution kernel was obtained based on backpropagated gradients. Then, the convolution kernels with large importance scales were activated to select target-aware features. Finally, the selected features were inputted into the SiamRPN module for target-background classification and the bounding box regression was applied to obtain an accurate bounding box of the target. Results of experiments on OTB2015 and VOT2018 datasets confirm that the proposed algorithm can achieve robust tracking of the target.

陈志旺, 张忠新, 宋娟, 罗红福, 彭勇. 基于目标感知特征筛选的孪生网络跟踪算法[J]. 光学学报, 2020, 40(9): 0915003. Zhiwang Chen, Zhongxin Zhang, Juan Song, Hongfu Luo, Yong Peng. Tracking Algorithm for Siamese Network Based on Target-Aware Feature Selection[J]. Acta Optica Sinica, 2020, 40(9): 0915003.

本文已被 4 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!