SLAM: A Lightweight Spatial Location Attention Module for Object Detection

Aiming to address the shortcomings of current object detection models, including a large number of parameters, the lack of accurate localization of target bounding boxes, and ineffective detection, this paper proposes a lightweight spatial location attention module (SLAM) that achieves adaptive adjustment of the attention weights of the location information in the feature map while greatly improving the feature representation capability of the network by learning the spatial location information in the input feature map. First, the SLAM module obtains the spatial distribution of the input feature map in the horizontal, vertical, and channel directions through the average pooling and maximum pooling operations, then generates the corresponding location attention weights by computing convolution and activation functions, and finally achieves the weighted feature map by aggregating the features along the three spatial directions respectively. Extensive experiments show that the SLAM module improves the detection performance of the model on the MS COCO dataset and the PASCAL VOC 2012 dataset with almost no additional computational overhead.