An Anchor-Free Vehicle Detection Algorithm in Aerial Image Based on Context Information and Transformer
Wangcheng Zhou, Jiaquan Shen, Ningzhong Liu, Shifeng Xia, Han Sun · IEEE Geoscience and Remote Sensing Letters · 2022
Vehicle detection in the aerial image is an essential and challenging task widely used in industry and agriculture. Deep learning technology has recently achieved rapid development and good object detection results. However, the background of aerial images is complex, targets are densely distributed, and some of them are occluded. For densely distributed targets, we need to predict at each feature point. In the case of complex background and target occlusion, it is often difficult to determine whether a location contains a target if the model only focuses on the local information. Therefore, we need a global perspective and contextual information to help train the model. This paper proposes a new anchor-free small object detection algorithm, which improves feature extraction by fusing contextual semantic information. In addition, a dynamic activation function is also used in our network, which helps us calculate the activation function value for each point from a global perspective. Moreover, we also use the channel attention module and the transformer as the spatial attention module to help the network efficiently obtain global information. We evaluate the effectiveness of our method on the public dataset DLR-3K and VEDAI, and the AP achieves 0.896 and 0.875.