Algorithm for Small Target Detection from an Unmanned Aerial Vehicle View

被引:0
|
作者
Zhang, Ruifang [1 ,2 ]
Du, Yiting [2 ]
Cheng, Xiaohui [3 ]
机构
[1] Guilin Univ Technol, Key Lab Adv Mfg & Automat Technol, Guilin 541006, Guangxi, Peoples R China
[2] Guilin Univ Technol, Coll Mech & Control Engn, Guilin 541006, Guangxi, Peoples R China
[3] Guilin Univ Technol, Coll Informat Sci & Engn, Guilin 541006, Guangxi, Peoples R China
关键词
YOLOv8s; ODE; BiEO-Neck; small object detection head; WIoU;
D O I
10.3788/LOP241149
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
This paper proposes a new bidirectional weighted multiscale dynamic approach, the BiEO-YOLOv8s algorithm, to enhance the detection of small targets in aerial images. It effectively addresses challenges such as complex backgrounds, large-scale variations, and dense targets. First, we design a new ODE module to replace certain C2f modules, enabling the accurate, quick, and multiangle location of target features. Then, we develop a bidirectional weighted multiscale dynamic neck network structure (BiEO-Neck) to achieve deep fusion of shallow and deep features. Second, adding a small object detection head further enhances feature extraction capability. Finally, the generalized intersection union ratio boundary loss function is used to replace the original boundary loss function, thereby enhancing the regression performance of the bounding box. Experiments conducted on the VisDrone dataset demonstrat that as compared to the base model YOLOv8s, the proposed model achieved a 6.1 percentage points improvement in mean average precision, with a detection speed of only 4. 9 ms. This performance surpasses that of other mainstream models. The algorithm effectiveness and adaptability are further confirmed through universality testing on the IRTarget dataset. The proposed algorithm can efficiently complete target detection tasks in of unmanned aerial vehicle aerial images.
引用
收藏
页数:9
相关论文
共 24 条
  • [1] Adaptive Sparse Convolutional Networks with Global Context Enhancement for Faster Object Detection on Drone Images
    Du, Bowei
    Huang, Yecheng
    Chen, Jiaxin
    Huang, Di
    [J]. 2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 13435 - 13444
  • [2] Lightweight Feature Fusion Network for Object Detection in Aerial Photography Images
    Fan Qiangqiang
    Shi Zaifeng
    Kong Fanning
    Li Shaoxiong
    Xiao Jun
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (10)
  • [3] He KM, 2020, IEEE T PATTERN ANAL, V42, P386, DOI [10.1109/ICCV.2017.322, 10.1109/TPAMI.2018.2844175]
  • [4] He Z F, 2022, CHINESE J LASERS, V49
  • [5] Liang X M, 2024, RADIO ENG
  • [6] Feature Pyramid Networks for Object Detection
    Lin, Tsung-Yi
    Dollar, Piotr
    Girshick, Ross
    He, Kaiming
    Hariharan, Bharath
    Belongie, Serge
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 936 - 944
  • [7] Path Aggregation Network for Instance Segmentation
    Liu, Shu
    Qi, Lu
    Qin, Haifang
    Shi, Jianping
    Jia, Jiaya
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 8759 - 8768
  • [8] Small Object Recognition Algorithm of Grain Pests Based on SSD Feature Fusion
    Lyu, Zongwang
    Jin, Huifang
    Zhen, Tong
    Sun, Fuyan
    Xu, Hui
    [J]. IEEE ACCESS, 2021, 9 : 43202 - 43213
  • [9] Ouyang Daliang, 2023, ICASSP 2023 - 2023 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), P1, DOI 10.1109/ICASSP49357.2023.10096516
  • [10] UAV Small Object Detection Algorithm Based on Context Information and Feature Refinement
    Peng, Yanfei
    Zhao, Tao
    Chen, Yankang
    Yuan, Xiaolong
    [J]. Computer Engineering and Applications, 2024, 60 (05) : 183 - 190