国内刊号:51-1307/TP
国际刊号:1001-9081
发布日期:
作者:于平平, 闫玉婷, 唐心亮, 苏鹤, 王建超
单位:1.河北科技大学 信息科学与工程学院,石家庄 050018;2.河北工业大学 电气工程学院,天津 300130
关键词:输电线路场景,目标检测,多目标跟踪,YOLOv5s,ByteTrack
基金:河北省教育厅青年基金资助项目(QN2023185)
在输电线路巡检任务中,采用深度学习技术实现施工机械运动的有效跟踪对智能电网建设至关重要。针对目标间遮挡干扰以及误检漏检造成的多目标跟踪性能显著下降的问题,提出一种改进YOLOv5s与优化ByteTrack相结合的多目标跟踪算法。在目标检测部分:首先,采用轻量级的Ghost卷积和SimAM构建SGC3 (SimAM and Ghost convolution with C3)模块,以提高特征利用率,并减少算法冗余计算;其次,在主干网络的深层,提出卷积引导的三重注意力模块R-Triplet(RFAConv with Triplet attention),从而利用多分支结构增强算法跨维度信息交互,并抑制不相关背景信息来提高目标的关联能力;最后,在特征融合部分添加多分支感受野模块(MRB),以利用空洞卷积扩大目标感受野,并增强多尺度目标全局特征信息的复用。在目标跟踪部分:在ByteTrack算法的基础上,根据施工机械的运动特点,提出一种自适应计算噪声尺度的NSA(Noise Scale Adaptively)卡尔曼滤波算法,以降低低质量检测框对滤波算法性能的影响;同时,在数据关联部分引入高斯平滑插值算法(GSI),从而进一步完善多目标跟踪的效果。实验结果表明,所提CRM-YOLOv5s算法的平均精度均值(mAP)达到了97.4%,与基线算法YOLOv5s相比提升了3.8个百分点,参数量和浮点运算量分别减少了0.28×106和1.8 GFLOPs,可见该算法在多种应用场景下的泛化能力更强。此外,相较于原YOLOv5s+ByteTrack跟踪算法,所提CRM-YOLOv5s算法与改进后的ByteTrack算法相结合后的多目标跟踪准确度(MOTA)提升了4.5个百分点,目标身份切换次数(IDs)减少了15,且获得了较高的推理速度,可见该算法适用于输电线路场景下施工机械的多目标跟踪任务。
来源:2025年第7期
《计算机应用》期刊编辑部