ABSSNet:基于注意力的空间分割网络用于交通场景理解

ABSSNet: Attention-Based Spatial Segmentation Network for Traffic Scene Understanding

IEEE Transactions on Cybernetics · 2021
被引 39
ABS 3

中文导读

提出一种结合注意力模块和空间CNN的神经网络,提升道路和车道线检测的准确性,减少误检和漏检,并构建了像素级道路分割数据集。

Abstract

The location information of road and lane lines is the supremely important thing for the automatic drive and auxiliary drive. The detection accuracy of these two elements dramatically affects the reliability and practicality of the whole system. In real applications, the traffic scene can be very complicated, which makes it particularly challenging to obtain the precise location of road and lane lines. Commonly used deep learning-based object detection models perform pretty well on the lane line and road detection tasks, but they still encounter false detection and missing detection frequently. Besides, existing convolution neural network (CNN) structures only pay attention to the information flow between layers, while it cannot fully utilize the spatial information inside the layers. To address those problems, we propose an attention-based spatial segmentation network for traffic scene understanding. We use the convolutional attention module to improve the network's understanding capacity of spatial location distribution. Spatial CNN (SCNN) obtains through the information flow within one single convolutional layer and improves the spatial relationship modeling ability of the network. The experimental results demonstrate that this method effectively improves the neural network's application ability of the spatial information, thereby improving the effect of traffic scene understanding. Furthermore, a pixel-level road segmentation dataset called NWPU Road Dataset is built to help improve the process of traffic scene understanding.

计算机视觉深度学习自动驾驶图像分割