SPMNet: A light-weighted network with separable pyramid module for real-time semantic segmentation

Abstract

ABSTRACT Real-time semantic segmentation aims to generate high-quality prediction in limited time. Recently, with the development of many related potential applications, such as autonomous driving, robot sensing and augmented reality devices, semantic segmentation is desirable to make a trade-off between accuracy and inference speed with limited computation resources. This paper introduces a novel effective and light-weighted network based on Separable Pyramid Module (SPM) to achieve competitive accuracy and inference speed with fewer parameters and computation. Our proposed SPM unit utilises factorised convolution and dilated convolution in the form of a feature pyramid to build a bottleneck structure, which extracts local and context information in a simple but effective way. Experiments on Cityscapes and Camvid datasets demonstrate our superior trade-off between speed and precision. Without pre-training or any additional processing, our SPMNet achieves 71.22% mIoU on Cityscapes test set at the speed of 94 FPS on a single GTX 1080Ti GPU card.