SqueezeDet: Unified, Small, Low Power Fully Convolutional Neural Networks for Real-Time Object Detection for Autonomous Driving

Bichen Wu, Forrest Iandola, Peter H. Jin, Kurt Keutzer; The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops, 2017, pp. 129-137

Abstract


Object detection is a crucial task for autonomous driving. In addition to requiring high accuracy to ensure safety, object detection for autonomous driving also requires real-time inference speed to guarantee prompt vehicle control, as well as small model size and energy efficiency to enable embedded system deployment. In this work, we propose SqueezeDet, a fully convolutional neural network for object detection that aims to simultaneously satisfy all of the above constraints. In our network, we use convolutional layers to extract feature maps and compute bounding boxes and class probabilities. The detection pipeline of our model only contains a single forward pass of a neural network, thus it is extremely fast. Our model is fully convolutional, which leads to small model size and better energy efficiency. Finally, our experiments show that our model is very accurate, achieving state-of-the-art accuracy on the KITTI benchmark. The source code of SqueezeDet is open-source released.

Related Material


[pdf] [arXiv]
[bibtex]
@InProceedings{Wu_2017_CVPR_Workshops,
author = {Wu, Bichen and Iandola, Forrest and Jin, Peter H. and Keutzer, Kurt},
title = {SqueezeDet: Unified, Small, Low Power Fully Convolutional Neural Networks for Real-Time Object Detection for Autonomous Driving},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
month = {July},
year = {2017}
}