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

Li Linhui,Yuan Shiwei,Lian Jing,Gu Tangpeng
DOI: https://doi.org/10.19620/j.cnki.1000-3703.20221220
2023-01-01
Abstract: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.
What problem does this paper attempt to address?