基于优化DeepSort的前方车辆多目标跟踪

Multi-target tracking of vehicles based on optimized DeepSort

为了提升自动驾驶汽车对周边环境的感知能力,提出优化DeepSort的前方多车辆目标跟踪算法.采用Gaussian YOLO v3作为前端目标检测器,基于DarkNet-53骨干网络训练,获得专门针对车辆的检测器Gaussian YOLO v3-vehicle,使车辆检测准确率提升3%.为了克服传统预训练模型没有针对车辆类别的缺点,提出采用扩增后的VeRi数据集进行重识别预训练.提出结合中心损失函数与交叉熵损失函数的新损失函数,使网络提取的目标特征有更好的类内聚合以及类间分辨能力.试验部分采集不同环境的实际道路视频,采用CLEAR MOT评价指标进行性能评估.结果表明,与基准DeepSort YOLO v3相比,跟踪准确度提升1%,身份切换次数减少4%.

A front multi-vehicle target tracking algorithm optimized by DeepSort was proposed in order to improve the awareness of autonomous vehicles to the surrounding environment.Gaussian YOLO v3 model was adopted as the front-end target detector,and training was based on DarkNet-53 backbone network.Gaussian YOLO v3-Vehicle,a detector specially designed for vehicles was obtained,which improved the vehicle detection accuracy by 3%.The augmented VeRi data set was proposed to conduct the re-recognition pre-training in order to overcome the shortcomings that the traditional pre-training model doesn't target vehicles.A new loss function combining the central loss function and the cross entropy loss function was proposed,which can make the target features extracted by the network become better in-class aggregation and inter-class resolution.Actual road videos in different environments were collected in the test part,and CLEAR MOT evaluation index was used for performance evaluation.Results showed a 1%increase in tracking accuracy and a 4%reduction in identity switching times compared with the benchmark DeepSort YOLO v3.