基于网络流跟踪的信号灯检测方法

作者：武悦; 陈海华^*; 于乔烽

来源：计算机应用研究, 2023, 1-9.

DOI：10.19734/j.issn.1001-3695.2023.07.0301

• 信号灯检测
• 帧间信息联合
• 多目标跟踪
• 费用流网络

结合信号灯信息对机动车行进速度进行引导，减少机动车启停次数，可有效减少废气排放，缓解其造成的污染问题。针对信号灯转换时刻的获取，该文提出了一种基于网络流跟踪的信号灯检测方法，在数据集中引入辅助信号灯类别进行训练，将视频序列中该类目标检测结果关联为踪片，并通过踪片建模多目标跟踪任务。该方法将多目标跟踪任务转换为最小费用流优化任务，以踪片作为节点建立最小费用流网络，提出了适合于信号灯的费用构建方式，通过最短路径算法求解，得到视频序列中辅助信号灯的多条轨迹。基于求解的轨迹结果和图像分类技术，最终实现信号灯检测性能的提升。该方法的跟踪性能相较于对比算法有大幅提升，并将小目标信号灯检测响应的mAP提升至94.35%。实验结果表明，基于网络流的建模方式可极大提升信号灯的跟踪准确率，结合跟踪轨迹还大幅提高了视频序列中小目标信号灯的检测准确率，并可有效确定信号灯状态的转换时刻。

• 单位
  南开大学