计算机应用

作者：张怡, 孙永荣, 赵科东, 李华, 曾庆化

单位：南京航空航天大学 导航研究中心，南京 210016

关键词：空中加油,检测跟踪一体化,网络轻量化,膨胀卷积,深度可分离卷积,网络优化

针对自主空中加油对接阶段的目标跟踪问题，提出一种空中加油场景下的目标联合检测跟踪算法。该算法采用检测跟踪一体化的CenterTrack网络实现对锥套的追踪，而针对计算量较大、训练耗时过长的问题，分别从模型设计与网络优化两方面改善该网络。首先，在跟踪器中引入膨胀卷积组，以在不改变感受野大小的前提下使得网络轻量化；同时，将输出部分的卷积层替换为深度可分离卷积层，从而减少网络的参数量与计算量；然后，对网络进行进一步的优化，即将随机梯度下降（SGD）法与Adam算法相结合，使网络更快收敛至稳定状态；最后，利用真实的空中加油场景视频与地面模拟视频制作相应格式的数据集，并将其用于实验验证。分别在自制的锥套数据集和MOT17公共数据集上进行了训练与测试，证实了提出算法的有效性。相较于原CenterTrack网络，改进的网络Tiny-CenterTrack减少了约48.6%的训练时长，并在实时性方面提升了8.8%。实验结果表明，改进后的网络在不损失网络性能的前提下可有效节省计算资源并在一定程度上提升实时性。

来源：2022年第9期

《计算机应用》期刊编辑部

查看计算机应用杂志2022年第9期