摘要
为了提升自动驾驶汽车对周边环境的感知能力,提出优化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.
作者
金立生
华强
郭柏苍
谢宪毅
闫福刚
武波涛
JIN Li-sheng;HUA Qiang;GUO Bai-cang;XIE Xian-yi;YAN Fu-gang;WU Bo-tao(School of Vehicle and Energy,Yanshan University,Qinhuangdao 066004,China;Hebei Key Laboratory of Special Delivery Equipment,Yanshan University,Qinhuangdao 066004,China;Transportation College,Jilin University,Changchun 130022,China;Department of Automotive Engineering,Hebei Institute of Mechanical and Electrical Technology,Xingtai 054000,China)
出处
《浙江大学学报(工学版)》
EI
CAS
CSCD
北大核心
2021年第6期1056-1064,共9页
Journal of Zhejiang University:Engineering Science
基金
国家重点研发计划资助项目(2018YFB1600501)
国家自然科学基金资助项目(52072333)
国家自然科学基金区域创新发展联合基金资助项目(U19A2069)
河北省省级科技计划资助项目(20310801D,E2020203092,F2021203107).