基于改进YOLOv7-DeepSort的红外视频多目标跟踪

摘要

针对红外图像纹理弱及多目标遮挡导致跟踪精度低的问题，构建了基于改进YOLOv7模型和多目标跟踪算法DeepSort的融合红外目标跟踪模型MSB-YOLOv7-DeepSort。采用SE(squeeze and excitation)通道注意力机制和双向特征金字塔网络提高红外目标的特征提取质量；利用轻量化网络MobileNetV3替换YOLOv7骨干网络，提升融合模型的推理速度。实验结果表明，MSB-YOLOv7-DeepSort模型在跟踪准确度、跟踪精确度、正确目标跟踪比例和帧率等方面均具有较好的性能。

关键词

红外目标跟踪 / YOLOv7 / 轻量化 / SE注意力机制 / MobileNetV3 / 双向特征金字塔网络