SSRNet: A Two-branch Real-time Semantic Segmentation Network for Road Scenes
Conference: ICMLCA 2021 - 2nd International Conference on Machine Learning and Computer Application
12/17/2021 - 12/19/2021 at Shenyang, China
Proceedings: ICMLCA 2021
Pages: 5Language: englishTyp: PDF
Personal VDE Members are entitled to a 10% discount on this title
Authors:
Shao, Huiwei; Zhao, Ji (School of Computer Science and Software Engineering, University of Science and Technology Liaoning, Anshan, China)
Abstract:
Semantic segmentation, one of the fundamental tasks in computer vision, requires classification for each pixel in an image, and thus semantic segmentation is time-consuming. With the rise of technologies such as autonomous driving, the realtime capability of semantic segmentation is becoming more and more important. In this paper, the authors propose a novel two-way real-time semantic segmentation network (SSRNet). We create an efficient feature extraction residual block using channel split and channel shuffle to balance operational efficiency with feature extraction and design a shared spatial path. To improve the segmentation performance of the model, we designed a feature fusion module with a skip connection. Evaluated on a single 1080Ti GPU, SSRNet can run at over 85 FPS and achieve 71.2% MIoU on the Cityscapes dataset.