计算机应用 ›› 2018, Vol. 38 ›› Issue (3): 666-670.DOI: 10.11772/j.issn.1001-9081.2017082025

• 人工智能 • 上一篇    下一篇

Faster R-CNN模型在车辆检测中的应用

王林, 张鹤鹤   

  1. 西安理工大学 自动化与信息工程学院, 西安 710048
  • 收稿日期:2017-08-21 修回日期:2017-10-16 出版日期:2018-03-10 发布日期:2018-03-07
  • 通讯作者: 张鹤鹤
  • 作者简介:王林(1962-),男,江苏东台人,教授,博士,主要研究方向:无线传感器网络、复杂网络社团发现、大数据、数据挖掘;张鹤鹤(1991-),女,河南灵宝人,硕士研究生,主要研究方向:大数据、深度学习。
  • 基金资助:
    陕西省科技计划重点项目(2017ZDCXL-GY-05-03)。

Application of Faster R-CNN model in vehicle detection

WANG Lin, ZHANG Hehe   

  1. College of Automation and Information Engineering, Xi'an University of Technology, Xi'an Shaanxi 710048, China
  • Received:2017-08-21 Revised:2017-10-16 Online:2018-03-10 Published:2018-03-07
  • Supported by:
    This work is partially supported by Shaanxi Province Science and Technology Key Project (2017ZDCXL-GY-05-03).

摘要: 针对传统机器学习方法在车辆检测应用中易受光照、目标尺度和图像质量等因素影响,效率低下且泛化能力较差的问题,提出一种基于改进的较快的基于区域卷积神经网络(R-CNN)模型的车辆检测方法。该方法以Faster R-CNN模型为基础,通过对输入图像进行卷积和池化等操作提取车辆特征,结合多尺度训练和难负样本挖掘策略降低复杂环境的影响,利用KITTI数据集对深度神经网络模型进行训练,并采集实际场景中的图像进行测试。仿真实验中,在保证检测时间的情况下,相对原Faster R-CNN算法检测精确度提高了约8%。实验结果表明,所提方法能够自动地提取车辆特征,解决了传统方法提取特征费时费力的问题,同时提高了车辆检测精确度,具有良好的泛化能力和适用范围。

关键词: 车辆检测, Faster R-CNN模型, 区域建议网络, 难负样本挖掘, KITTI数据集

Abstract: Since the traditional machine learning methods are easy to be affected by light, target scale and image quality in vehicle detection applications, resulting the low efficiency and generalization ability, a vehicle detection method based on improved Faster Regions with Convolutional Neural Network features (R-CNN) model was proposed. On the basis of Faster R-CNN model, through convolution and pooling operations to extract the features of vehicles, by combining with multi-scale training and hard negative sample mining strategy to reduce the influence of complex environment, the KITTI data set was used to train the deep neural network model, and the images were collected from actual scene to test the trained neural network model. In the simulation experiments, while the detection time was guaranteed, the detection accuracy of the proposed method was improved by about 8% compared to the original Faster R-CNN algorithm. The experimental results show that the proposed method can automatically extract the features of vehicles, solve the time-consuming and laborious problem of extracting features by traditional methods, effectively improve the accuracy of vehicle detection, and has good generalization ability and wide range of applications.

Key words: vehicle detection, Faster Regions with Convolutional Neural Network features (R-CNN) model, region proposal network, hard negative sample mining, KITTI data set

中图分类号: