Drone Target Detection Algorithm Based on Multi-scale Fusion and Lightweight Network

被引：0
|
作者：
Xue S. [1 ,2 ]
Lu T. [1 ]
Lü Q. [1 ]
Cao G. [2 ]
机构：
[1] College of Mechanical and Electrical Engineering, Changchun University of Science and Technology, Changchun
[2] Chongqing Research Institute, Changchun University of Science and Technology, Chongqing
来源：
Hunan Daxue Xuebao/Journal of Hunan University Natural Sciences | 2023年 / 50卷 / 08期
关键词：
clustering optimization method; coordinated attention mechanism; feature fusion; object detection; scale change;
D O I：
10.16339/j.cnki.hdxbzkb.2023286
中图分类号：
学科分类号：
摘要：
Aiming at the difficulty of real-time detection and limited computing resources due to the scale change of drones in public safety areas such as playgrounds and parks，a network dynamic real-time detection method for drones，YOLO-Ads，is proposed to increase the robustness of network ability to detect drone change. Firstly，the drone data set was built independently. Secondly，a new MDDRDNet network was established with the lightweight network as the backbone to reduce the complexity of model calculation，and the coordinated attention mechanism module was introduced to strengthen the network’s attention to space and channels. Then，the mean clustering algorithm is used to regenerate the prior frame，and the optimization method combining multiple probes and multiple data sets is used in the selection of the prior frame，so that the regenerated prior frame matches the drone better. The idea of feature fusion and residual error establishes a new detector head to adapt to the detection of smaller-scale drones. Finally，a class activation mapping module is introduced into the detection module to generate a heat map，so as to observe the sensitivity of the network to changes in the scale of drones. At the same time，comparative experiments are conducted with the current mainstream networks SSD，CenterNet，YOLOv5，YOLOx，etc.，and different backbone networks ResNet，EfficientNet，VGGNet，etc. The experimental results show that the newly proposed algorithm has an average accuracy of 96.62% in the detection of scale-changing drones. Compared with the YOLOv4 algorithm，it is increased by 1.88% . The detection speed is 47 frames per second，which is 19 frames higher than that of the YOLOv4 algorithm. The memory occupied by the model is about 10.844 M，which is about one-sixth of the original memory. It reflects the timeliness and robustness of the method. © 2023 Hunan University. All rights reserved.
引用
收藏
页码：82 / 93
页数：11
相关论文
共 37 条
  • [1] XUE S, WEI L W，, GU C Y, Drone identification method based on mixed domain attention mechanism［J］, Journal of Xi’ an Jiaotong University, 56, 10, pp. 141-150, (2022)
  • [2] WEI P W．, Design of real-time detection and tracking of anti-drone system based on sky background［D］, pp. 24-27, (2021)
  • [3] XUE S, Image recognition method of anti drone system based on coordinate attention mechanism［J］, Infrared and Laser Engineering, 51, 9, pp. 417-427, (2022)
  • [4] CAI Z H, CHEN W J, ZHAO J，, Et al., Object detection and obstacle avoidance based on dynamic vision sensor for UAV［J/OL］, Journal of Beijing University of Aeronautics and Astronautics
  • [5] GIRSHICK R，, DONAHUE J，, DARRELL T, Region-based convolutional networks for accurate object detection and segmentation, IEEE Transactions on Pattern Analysis and Machine Intelligence, 38, 1, pp. 142-158, (2016)
  • [6] SSD：single shot multibox detector, (2015)
  • [7] Proceedings of the 2016 IEEE Computer Society Conference on Computer Vision and Pattern Recognition, pp. 779-788, (2016)
  • [8] KOLTUN V．, Multi-scale context aggregation by dilated convolutions, (2015)
  • [9] WANG Y D, WANG T B．, Algorithm for foggy-image pedestrian and vehicle detection ［J］, Journal of Xidian University, 47, 4, pp. 70-77, (2020)
  • [10] GUO Z K，, TIAN L，, HAN N, Radar HRRP based few-shot target recognition with CNN-SSD ［J］, Journal of Xidian University, 48, 2, pp. 7-14, (2021)
1234