A Lightweight Uav Swarm Detection Method Integrated Attention Mechanism
Author:
Wang ChuanyunORCID, Meng Linlin, Gao QianORCID, Wang Jingjing, Wang Tian, Liu Xiaona, Du Furui, Wang Linlin, Wang Ershen
Abstract
Aiming at the problems of low detection accuracy and large computing resource consumption of existing Unmanned Aerial Vehicle (UAV) detection algorithms for anti-UAV, this paper proposes a lightweight UAV swarm detection method based on You Only Look Once Version X (YOLOX). This method uses depthwise separable convolution to simplify and optimize the network, and greatly simplifies the total parameters, while the accuracy is only partially reduced. Meanwhile, a Squeeze-and-Extraction (SE) module is introduced into the backbone to improve the model′s ability to extract features; the introduction of a Convolutional Block Attention Module (CBAM) in the feature fusion network makes the network pay more attention to important features and suppress unnecessary features. Furthermore, Distance-IoU (DIoU) is used to replace Intersection over Union (IoU) to calculate the regression loss for model optimization, and data augmentation technology is used to expand the dataset to achieve a better detection effect. The experimental results show that the mean Average Precision (mAP) of the proposed method reaches 82.32%, approximately 2% higher than the baseline model, while the number of parameters is only about 1/10th of that of YOLOX-S, with the size of 3.85 MB. The proposed approach is, thus, a lightweight model with high detection accuracy and suitable for various edge computing devices.
Funder
National Natural Science Foundation of China Scientific Research Program of Liaoning Provincial Education Department of China Young and middle-aged Science and Technology Innovation Talents Project of Shenyang of China Doctoral Scientific Research Foundation of Shenyang Aerospace University
Subject
Artificial Intelligence,Computer Science Applications,Aerospace Engineering,Information Systems,Control and Systems Engineering
Reference36 articles.
1. Tang, J., Duan, H., and Lao, S. (2022). Swarm intelligence algorithms for multiple unmanned aerial vehicles collaboration: A comprehensive review. Artif. Intell. Rev., 1–33. 2. Key Technologies and Development Trend of UAV Swarm Operation;Zhang;China New Telecommun.,2022 3. Exploration of UAV cluster defense technology;Cai;Aerodyn. Missile J.,2020 4. Andrew, G., Zhu, M., Chen, B., Kalenichenko, D., Wang, W., Weyand, T., Andreetto, M., and Adam, H. (2017). MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications. arXiV. 5. Wang, C., Su, Y., Wang, J., Wang, T., and Gao, Q. (2022). UAVSwarm Dataset: An Unmanned Aerial Vehicle Swarm Dataset for Multiple Object Tracking. Remote Sens., 14.
Cited by
3 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
|
|