计算机工程与应用 ›› 2019, Vol. 55 ›› Issue (13): 225-230.DOI: 10.3778/j.issn.1002-8331.1901-0055

• 工程与应用 • 上一篇    下一篇

改进的RetinaNet模型的车辆目标检测

宋欢欢,惠  飞,景首才,郭兰英,马峻岩   

  1. 长安大学 信息工程学院,西安 710064
  • 出版日期:2019-07-01 发布日期:2019-07-01

Improved RetinaNet Model for Vehicle Target Detection

SONG Huanhuan, HUI Fei, JING Shoucai, GUO Lanying, MA Junyan   

  1. School of Information and Engineering, Changan University, Xi’an 710064, China
  • Online:2019-07-01 Published:2019-07-01

摘要: 目前,在智能交通领域使用深度学习方法进行车辆目标检测已成为研究热点。针对传统机器学习方法的性能易受光照、角度、图像质量等外界因素影响,检测步骤繁琐等问题,通过对当下经典的一阶目标检测模型和二阶目标检测模型进行分析,提出了一种基于改进的一阶目标检测模型RetinaNet的车辆目标检测方法,使用深度残差网络自主获取图像特征,融合MobileNet网络结构进行模型加速,把复杂交通场景下的目标检测问题转化为车辆类型的三分类问题,利用KITTI数据集进行训练,并使用实际场景中的图像进行测试。实验结果表明,改进的RetinaNet模型在保证检测时间的情况下,相比原RetinaNet模型MAP值提高了2.2个百分点。

关键词: 深度学习, 交通场景, 车辆检测, 深度残差网络

Abstract: At present, in the field of intelligent transportation, vehicle target detection has become a research hotspot by using deep learning method. As traditional machine learning method is easily affected by external factors such as illumination, angle and image quality, and the cumbersome detection steps, current one-stage target detection models and two-stage target detection models are analyzed, and vehicle target detection method is proposed by on one-stage target detection model and is named RetinaNet, which uses deep residual network to acquire image features autonomously, integrates MobileNet network structure to accelerate the model and transforms the target detection problem in complex traffic scenarios into the three-category problem of vehicle type. KITTI data set is adopted to train and actual scenes images are applied to test. Experimental results show that the proposed method improves the MAP value by 2.2 percentage points in comparison with original RetinaNet model.

Key words: deep learning, traffic scenes, vehicle detection, deep residual network