基于多维注意力融合的驾驶场景分割增强算法.

To address the problem of unbalanced computational resource consumption and accuracy of semantic segmentation models using attention mechanism, this paper proposed a lightweight attention enhancement algorithm for semantic segmentation. Firstly, it designed a striped dimensional attention mechanism based on the shape characteristics of objects in driving scenes, used striped pooling instead of traditional square convolution, and combined dimensionality reduction operations to extract long-range semantic associations in each dimension to cut down the model computation. Then it fused the attention on channel domain and spatial domain to form a lightweight multidimensional attention fusion module that could be superimposed and disassembled to extract feature information in all directions and further improve the model accuracy. Finally, it inserted the module into the ResNet-101 backbone based encoding-decoding network to guide the semantic fusion of high and low layers, correct the feature map edge information, and supplement the prediction details. The experiments show that the module has strong robustness and generalization ability, cutting about 90% of the number of parameters and 80% of the computation compared with the same type of attention mechanism, and the segmentation accuracy still achieves a stable improvement. [ABSTRACT FROM AUTHOR]