1.一种基于选择性状态空间的目标跟踪方法,其特征在于,包括以下步骤:S1:获取模板区域图像和搜索区域图像,分别提取特征后,进行拼接,得到待处理特征;
待处理特征经线性变换生成第一查询矩阵、第一键矩阵和第一值矩阵,第一查询矩阵进行池化操作后,生成代理令牌;
利用代理令牌与第一键矩阵的点积及位置偏置,计算代理注意力权重,代理注意力权重对第一值矩阵进行加权,得到代理特征;
利用第一查询矩阵与代理令牌的点积及代理偏置,计算交互特征注意力权重,交互特征注意力权重与代理特征进行点积操作,得到交互特征;
S2:获取当前帧前一帧的状态,结合交互特征和当前帧的可学习矩阵,根据公式:,构建选择性状态空间并得到当前帧的状态;式中, 为当前帧的状态; 为当前帧前一帧的状态; 为交互特征; 为当前帧的可学习矩阵; 为状态转移矩阵;
S3:根据公式: ,当前帧的状态与投影矩阵进行结合,生成目标特征;式中,为目标特征; 为投影矩阵; 为当前帧的状态;
目标特征经自注意力机制处理后,得到融合特征;
融合特征经线性变换,得到边界框坐标,进行目标跟踪。
2.根据权利要求1所述的基于选择性状态空间的目标跟踪方法,其特征在于,步骤S3中,所述融合特征经线性变换,得到边界框坐标,具体为:根据公式: ,得到边界框坐标;
式中, 为边界框坐标; 为权重矩阵; 为融合特征; 为偏置项。
3.根据权利要求1所述的基于选择性状态空间的目标跟踪方法,其特征在于,步骤S3中,所述目标特征经自注意力机制处理后,得到融合特征,具体为:目标特征经线性变换生成第二查询矩阵、第二键矩阵和第二值矩阵,第二查询矩阵和第二键矩阵进行点积操作,得到注意力权重,注意力权重对第二值矩阵进行加权求和,得到融合特征。
4.根据权利要求1所述的基于选择性状态空间的目标跟踪方法,其特征在于,步骤S3中,得到边界框坐标后,还包括对边界框坐标进行非极大值抑制处理。
5.一种基于选择性状态空间的目标跟踪系统,其特征在于,包括:特征处理模块:用于获取模板区域图像和搜索区域图像,分别提取特征后,进行拼接,得到待处理特征;
待处理特征经线性变换生成第一查询矩阵、第一键矩阵和第一值矩阵,第一查询矩阵进行池化操作后,生成代理令牌;
利用代理令牌与第一键矩阵的点积及位置偏置,计算代理注意力权重,代理注意力权重对第一值矩阵进行加权,得到代理特征;
利用第一查询矩阵与代理令牌的点积及代理偏置,计算交互特征注意力权重,交互特征注意力权重与代理特征进行点积操作,得到交互特征;
选择性状态空间构建模块:用于获取当前帧前一帧的状态,结合交互特征和当前帧的可学习矩阵,根据公式: ,构建选择性状态空间并得到当前帧的状态;式中, 为当前帧的状态; 为当前帧前一帧的状态; 为交互特征; 为当前帧的可学习矩阵; 为状态转移矩阵;
目标跟踪模块:用于根据公式: ,当前帧的状态与投影矩阵进行结合,生成目标特征;式中, 为目标特征; 为投影矩阵; 为当前帧的状态;
目标特征经自注意力机制处理后,得到融合特征;
融合特征经线性变换,得到边界框坐标,进行目标跟踪。
6.一种基于选择性状态空间的目标跟踪装置,其特征在于,包括处理器和存储器,其中,所述处理器执行所述存储器中保存的计算机程序时实现如权利要求1‑4中任一项所述的基于选择性状态空间的目标跟踪方法。
7.一种基于选择性状态空间的目标跟踪介质,其特征在于,用于存储计算机程序,其中,所述计算机程序被处理器执行时实现如权利要求1‑4中任一项所述的基于选择性状态空间的目标跟踪方法。