基于协作注意力和图神经网络的轻量化车位检测算法

Lightweight Parking-Slot Detection Algorithm Based on Collaborative Attention and Graph Neural Network

为提高自动泊车过程车位检测的实时性和准确性,提出一种基于协作注意力和图神经网络的轻量化车位检测算法。首先,采用轻量化的网络结构,以改进的MobileNetV3作为特征提取网络,通过深度可分离卷积获得车位标记点的位置信息和特征信息,并将二者结合得到标记点的融合特征,然后构建图神经网络结构以增强车位标记点之间的内在联系,并结合协作注意力机制实现对多头注意力的整合,最后,通过公共车位数据集PS2.0对所提出的算法进行测试,结果表明,该算法的检测精度优于当前主流算法,平均每帧图像推理时间可缩短至10.1ms,具备良好的准确性和实时性。

In order to improve the real-time and accuracy of parking slot detection in automatic parking,this paper proposed a lightweight parking-slot detection algorithm based on collaborative attention and graph neural network.Firstly,This algorithm used a lightweight network structure and the improved MobileNetV3 as the feature extraction network,obtained the location information and feature information of the parking-slot marker points through depthwise separable convolution,combined them to obtain the fused features of the marker points,then constructed a graph network structure to enhance the internal relationship of the parking-slot marker points,and combined the cooperative attention mechanism to integrate multiple attention.Finally,the algorithm was tested on the public parking-slot dataset PS2.0.The results indicate that the detection accuracy is better than the current mainstream algorithm,the average reasoning speed of each frame of image can reach 10.1 ms,with good accuracy and real-time performance.