基于注意力机制与孪生网络的目标跟踪方法及相关设备

摘要

本发明提供了一种基于注意力机制与孪生网络的目标跟踪方法及相关设备，通过将目标模板和搜索区域输入至预设的目标跟踪模型，通过预设的目标跟踪模型输出目标模板在所述搜索区域中的目标跟踪信息；所述目标跟踪模型中包含孪生网络，并且孪生网络中增加设置通道注意力模块和/或空间注意力模块。本实施例提供的目标跟踪方法及相关设备，由于在残差网络中加入通道注意力模块和/或空间注意力模块，显著提高了孪生跟踪算法的平均期望重叠率和鲁棒性，改善跟踪中的运动变化、相机运动、遮挡、尺寸变化属性的鲁棒性，因此使用本实施例提供的目标跟踪方法进行目标跟踪预测时，可以获取到较准确的结果。