Tiny YOLO Optimization Oriented Bus Passenger Object Detection
Shuo Zhang, Yanxia Wu, Chaoguang Men, Xiaosong Li
Abstract
The real-time collection of bus passenger object detection is an essential part of developing a smart bus system. The difficulty of object detection mainly lies in the objective factors, such as: clothing, hair style and accessories, light, etc. Traditional object detection methods with the artificial feature extraction suffers from insufficient strength in expression, generalization, and recognition rate. The object detection method based on deep learning mainly uses the convolutional neural network in deep learning to learn features from a large set of data. The learned features can describe the rich information inherent in the data, and improve the expression ability of the features as well as the recognition accuracy. Due to too many parameters of the Convolutional neural network (CNN) model, the amount of calculation is too large to be operated on the vehicle terminal. To reduce calculation burden and improve the operation speed, we employs the depthwise separable convolution method to optimize the convolutional layer of tiny YOLO network model. It decomposes a complete convolution operation into depthwise convolution and pointwise convolution, thus reducing the parameter amount of the CNN and improving the operation speed. The experiment results reveal that the speed of bus passenger object detection detected by our improved model is 4 times faster than the previous one but with the nearly same detection accuracy.