Vehicle road detection based on pyramid network and self-attention mechanism

Aiming at the imbalance between the accuracy and real-time performance of lane detection, this paper proposes a novel vehicle road detection based on pyramid network and self-attention mechanism (namely PFSA). In this method, a two-sided multi-scale fusion network is used to realise the information exchange between shallow features and deep features, and to obtain contextual semantics. A new asymmetric convolution pyramid module is proposed, which fuses asymmetric convolution into cavity convolution layers with different expansion rates to improve feature extraction capability and reduce computation. A two-stage training method was used to train on a public data set and compared with other advanced lane detection methods. The experimental results show that the accuracy is 98.3% and the detection speed is 18.391 frames per second (fps), which is much better than other lane detection methods.