GBCNN: A Full GPU-Based Batch Multi-Task Cascaded Convolutional Networks

Abstract

Recently, the face detection and alignment is so popular and widely used in many research and application fields. Many superior face detection algorithms such as multi-task cascade convolutional network have been presented. However, it has difficulty in predicting faces among the big scale images in real time due to its three stages cascade architecture with less optimization. In this paper, we propose a full GPU-based batch multi-task cascade convolutional network which is carefully designed and optimized in each step to gain a superior speed performance. In addition, we present a novel parallel memory allocation strategy, which further enables our algorithm to support the batch operation, so that the system throughput increases significantly. In the experiment, our method achieves up to 300fps, over 600% speedup with an equal accuracy over the state-of-the-art methods on the face detection benchmarks.

Keywords

• CNN
• GPU
• face detection