Laser & Optoelectronics Progress, Volume. 56, Issue 18, 181003(2019)

Vehicle Detection Algorithm Based on Convolutional Neural Network and RGB-D Images

Decheng Wang1, Xiangning Chen2、*, Feng Zhao1,3, and Haoran Sun4
Author Affiliations
  • 1 Graduate School, Space Engineering University, Beijing 101416, China
  • 2 School of Space Information, Space Engineering University, Beijing 101416, China
  • 3 61618 Troops, Beijing 100094, China
  • 4 Jiuquan Satellite Launch Centre, Jiuquan, Gansu 730000, China
  • show less

    Aim

    ing at the problem that using RGB images for vehicle detection are affected by complex conditions such as road shadow, vehicle reflection and insufficient light. The paper proposes a vehicle detection algorithm based on convolutional neural network and combination of RGB and depth images. Two improved models of single-channel RG-D and double-channel RGB-D fusion networks are designed to improve detection speed and accuracy respectively. The algorithm is tested with (Grand Theft Auto) vehicle dataset and compared with other popular algorithms based on RGB images. The results show that compared with Yolo v2 algorithm based on RGB images, detection accuracy and recall rates increase 5.69% and 6.31% respectively by double-channel RGB-D fusion network, and the fastest detection speed of single image reaches 24 ms with single-channel RG-D fusion network. Experiments show that the improved network model based on RGB-D images can achieve real-time detection and effectively improve vehicle detection accuracy.

    Tools

    Get Citation

    Copy Citation Text

    Decheng Wang, Xiangning Chen, Feng Zhao, Haoran Sun. Vehicle Detection Algorithm Based on Convolutional Neural Network and RGB-D Images[J]. Laser & Optoelectronics Progress, 2019, 56(18): 181003

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Feb. 25, 2019

    Accepted: Apr. 1, 2019

    Published Online: Sep. 9, 2019

    The Author Email: Chen Xiangning (18810836867@163.com)

    DOI:10.3788/LOP56.181003

    Topics