Research on Visual Target Detection Method for Smart City Unmanned Aerial Vehicles Based on Transformer

Unmanned aerial vehicles play a significant role in the automated inspection of future smart cities, which can ensure the safety of urban residents’ lives and property and the normal operation of the city. However, there may be situations where small targets in drone images are difficult to detect and the detection is unclear when the targets are similar to the environment. In response to the above problems, this paper proposes a real-time target detection method for unmanned aerial vehicle images based on Transformer. Aiming at the problem of small targets lacking visual features, a feature fusion module was designed, which realizes the interaction and fusion of features at different levels and improves the feature expression ability of small targets. Aiming at the problem of discontinuous features when the target is similar to the environment, a multi-head attention algorithm based on Transformer is designed. By extracting the context information of the target, the recognition ability of targets similar to the environment is improved. On the target image dataset collected by unmanned aerial vehicles in smart cities, the detection accuracy of the method described in this paper has reached 85.9%.