Small Object Recognition Algorithm Based on Hybrid Control and Feature Fusion

被引:1
|
作者
Zhu, Gaofeng [1 ,2 ]
Wang, Zhixue [1 ]
Zhu, Fenghua [3 ,4 ]
Xiong, Gang [4 ]
Li, Zheng [1 ]
机构
[1] Shandong Jiaotong Univ, Sch Rail Transit, Jinan 250300, Shandong, Peoples R China
[2] CASP Co, AI Res Ctr, Suzhou 215200, Jiangsu, Peoples R China
[3] CASP Co, Suzhou 215200, Jiangsu, Peoples R China
[4] Chinese Acad Sci, Inst Automat, State Key Lab Multimodal Artificial Intelligence S, Beijing 100190, Peoples R China
关键词
Attention mechanism; drone; deep learning; object detection;
D O I
10.1109/JRFID.2024.3384483
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Drone detection plays a key role in various fields, but from the perspective of drones, factors such as the size of the target, interference from different backgrounds, and lighting affect the detection effect, which can easily lead to missed detections and false detections. To address this problem, this paper proposes a small target detection algorithm. First, the hybrid control of attention mechanism and a convolutional module (HCAC) are used to effectively extract contextual details of targets of different scales, directions, and shapes, while relative position encoding is used to associate targets with position information. Secondly, in view of the small size characteristics of small targets, a high-resolution detection branch is introduced, the large target detection head and its redundant network layers are pruned, and a multi-level weighted feature fusion network (MWFN) is used for multi-dimensional fusion. Finally, the WIoU loss is used as a bounding box regression loss, combined with a dynamic non-monotonic focusing mechanism, to evaluate the quality of anchor boxes so that the detector can handle anchor boxes of different qualities, thus improving the overall performance. Experiments were conducted on the UAV aerial photography data set VisDrone2019. The results showed that the accuracy of P increased by 9.0% and MAP by 9.8%, with higher detection results.
引用
收藏
页码:484 / 492
页数:9
相关论文
共 50 条
  • [41] Facial Expression Recognition Algorithm Based on CNN and LBP Feature Fusion
    Yang, Xinli
    Li, Ming
    Zhao, ShiLin
    PROCEEDINGS OF 2017 INTERNATIONAL CONFERENCE ON ROBOTICS AND ARTIFICIAL INTELLIGENCE (ICRAI 2017), 2015, : 33 - 38
  • [42] Feature Fusion Expression Recognition Algorithm Based on DCA Dimensionality Reduction
    Wang, Xiaofang
    Fang, Dengjie
    Zou, Qianying
    Shi, Yifei
    GREEN, PERVASIVE, AND CLOUD COMPUTING, GPC 2022, 2023, 13744 : 17 - 34
  • [43] A Research on the Fruit Recognition Algorithm Based on the Multi-Feature Fusion
    Tang, Yanfeng
    Zhang, Yawan
    Zhu, Ying
    2020 5TH INTERNATIONAL CONFERENCE ON MECHANICAL, CONTROL AND COMPUTER ENGINEERING (ICMCCE 2020), 2020, : 1865 - 1869
  • [44] A Mandarin Tone Recognition Algorithm Based on Random Forest and Feature Fusion
    Yan, Jiameng
    Meng, Qiang
    Tian, Lan
    Wang, Xiaoyu
    Liu, Junhui
    Li, Meng
    Zeng, Ming
    Xu, Huifang
    MATHEMATICS, 2023, 11 (08)
  • [45] A small object detection algorithm based on feature interaction and guided learning
    Shao, Xiang-Ying
    Guo, Ying
    Wang, You-Wei
    Bao, Zheng-Wei
    Wang, Ji-Yu
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2024, 98
  • [46] Object Tracking Algorithm Based on Dual Color Feature Fusion with Dimension Reduction
    Hu, Shuo
    Ge, Yanan
    Han, Jianglong
    Zhang, Xuguang
    SENSORS, 2019, 19 (01)
  • [47] Object Detection Algorithm for Complex Road Scenes Based on Adaptive Feature Fusion
    Ran, Xiansheng
    Su, Shanjie
    Chen, Junhao
    Zhang, Zhiyun
    Computer Engineering and Applications, 2023, 59 (24) : 216 - 226
  • [48] An improved UPF object tracking algorithm based on multi-feature fusion
    Li, Xiao-Xu
    Dai, Bin
    Cao, Jie
    Shanghai Jiaotong Daxue Xuebao/Journal of Shanghai Jiaotong University, 2014, 48 (10): : 1473 - 1478
  • [49] Multi feature fusion-based water occlusion object detection algorithm
    Feng H.
    Jiang C.
    Xu H.
    Xie L.
    Huazhong Keji Daxue Xuebao (Ziran Kexue Ban)/Journal of Huazhong University of Science and Technology (Natural Science Edition), 2024, 52 (04): : 76 - 81
  • [50] Object Tracking Algorithm Based on Correlation-Transformer Dual Feature Fusion
    Li Z.
    Fan H.
    Wang Q.
    Yu Z.
    Tang Y.
    Jiqiren/Robot, 2024, 46 (01): : 16 - 26