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

作     者：宫华 张众垚 胡雨桐 刘芳 GONG Hua;ZHANG Zhongyao;HU Yutong;LIU Fang

作者机构：沈阳理工大学理学院沈阳110159 沈阳理工大学自动化与电气工程学院沈阳110159 辽宁省兵器工业智能优化与控制重点实验室沈阳110159 电磁空间安全国家重点实验室天津300308 

出 版 物：《沈阳理工大学学报》 (Journal of Shenyang Ligong University)

年 卷 期：2024年第43卷第6期

页      面：20-27页

学科分类：1305[艺术学-设计学（可授艺术学、工学学位）] 13[艺术学] 081104[工学-模式识别与智能系统] 08[工学] 0804[工学-仪器科学与技术] 081101[工学-控制理论与控制工程] 0811[工学-控制科学与工程] 

基　　金：军委科技委重点实验室预研基金项目(2022JCJQLB055008) 

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

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