一种基于改进YOLO V7和Deep-Sort的交通流量统计方法

作者：朱隽洋; 姬红兵; 张文博; 李林; 臧博

来源：2023-04-13, 中国, CN202310392311.6.

本发明公开了一种基于改进YOLO V7和Deep-Sort的交通流量统计方法,包括以下步骤；(1)准备车辆数据集；(2)搭建改进的YOLOv7模型并进行训练,用于对步骤(1)的车辆数据集检测；改进的YOLOv7模型包括输入端、特征提取网络、特征融合网络和输出端；(3)搭建Deep-Sort模型对检测后的车辆跟踪；Deep-Sort模型包括目标检测模块、位置预测模块、特征匹配模块和更新模块；(4)使用基于运动轨迹与检测线的交通流量统计方法获取交通监控视频、绘制虚拟检测线、设置感兴趣区域、进入检测流程和跟踪流程,从而完成交通流量统计。本发明能够提高目标车辆的检测精度和提高车辆跟踪速度。

• 单位
  西安电子科技大学