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 条
  • [21] Design of UAV Single Object Tracking Algorithm Based on Feature Fusion
    Li, Changze
    Liu, Xiaoxiong
    Zhang, Xingwang
    Qin, Bin
    2021 PROCEEDINGS OF THE 40TH CHINESE CONTROL CONFERENCE (CCC), 2021, : 3088 - 3092
  • [22] A new particle filter object tracking algorithm based on feature fusion
    Yan, He
    Liu, Jie
    Yang, De-Hong
    Wang, Pu
    Jin, Wei
    Guangdianzi Jiguang/Journal of Optoelectronics Laser, 2014, 25 (10): : 1990 - 1999
  • [23] Algorithm of Moving Object Detection based on Multi-feature Fusion
    Cao, Jianrong
    Sun, Xuemei
    Zhao, Shusheng
    Wang, Yameng
    Gong, Shulan
    2017 IEEE INTERNATIONAL CONFERENCE ON INFORMATION AND AUTOMATION (IEEE ICIA 2017), 2017, : 931 - 935
  • [24] Underwater object detection algorithm based on channel attention and feature fusion
    Zhang Y.
    Li X.
    Sun Y.
    Liu S.
    Xibei Gongye Daxue Xuebao/Journal of Northwestern Polytechnical University, 2022, 40 (02): : 433 - 441
  • [25] Moving Object Tracking Algorithm Based on Depth Feature Adaptive Fusion
    Yang Rui
    Zhang Baohua
    Zhang Yanyue
    Lu Xiaoqi
    Gu Yu
    Wang Yueming
    Liu Xin
    Ren Yan
    Li Jianjun
    LASER & OPTOELECTRONICS PROGRESS, 2020, 57 (18)
  • [26] Small object detection model based on feature fusion of attention mechanism
    Chen H.
    Zhen X.
    Zhao T.
    Huazhong Keji Daxue Xuebao (Ziran Kexue Ban)/Journal of Huazhong University of Science and Technology (Natural Science Edition), 2023, 51 (03): : 60 - 66
  • [27] Small Object Detection in Traffic Scenes Based on Attention Feature Fusion
    Lian, Jing
    Yin, Yuhang
    Li, Linhui
    Wang, Zhenghao
    Zhou, Yafu
    SENSORS, 2021, 21 (09)
  • [28] Research on Small Object Detection Based on Feature Fusion and Attention Mechanism
    Liu, Jianwei
    Liu, Zheng
    Lu, Jingwen
    Li, Chuancan
    Chen, Gangqiang
    39TH YOUTH ACADEMIC ANNUAL CONFERENCE OF CHINESE ASSOCIATION OF AUTOMATION, YAC 2024, 2024, : 2285 - 2291
  • [29] Adaptive Feature Fusion for Small Object Detection
    Zhang, Qi
    Zhang, Hongying
    Lu, Xiuwen
    APPLIED SCIENCES-BASEL, 2022, 12 (22):
  • [30] The object recognition algorithm based on local gradient ratio feature measurement
    Chen, Yuantao
    Zuo, Jingwen
    She, Kang
    Xiang, Zhiwu
    Wang, Zhongyuan
    Journal of Information and Computational Science, 2015, 12 (12): : 4689 - 4695