YOLO Adaptive Developments in Complex Natural Environments for Tiny Object Detection

被引:3
|
作者
Zhong, Jikun [1 ]
Cheng, Qing [1 ,2 ]
Hu, Xingchen [1 ]
Liu, Zhong [1 ,2 ]
机构
[1] Natl Univ Def Technol, Lab Big Data & Decis, 109 Deya Rd, Changsha 410003, Peoples R China
[2] Hunan Inst Adv Technol, 699 Qingshan Rd, Changsha 410003, Peoples R China
基金
中国国家自然科学基金;
关键词
object detection; complex natural environment; YOLOV5s; copy-paste; self-attention mechanism; head network; backbone network;
D O I
10.3390/electronics13132525
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Detection of tiny object in complex environments is a matter of urgency, not only because of the high real-world demand, but also the high deployment and real-time requirements. Although many current single-stage algorithms have good detection performance under low computing power requirements, there are still significant challenges such as distinguishing the background from object features and extracting small-scale target features in complex natural environments. To address this, we first created real datasets based on natural environments and improved dataset diversity using a combination of copy-paste enhancement and multiple image enhancement techniques. As for the choice of network, we chose YOLOV5s due to its nature of fewer parameters and easier deployment in the same class of models. Most improvement strategies to boost detection performance claim to improve the performance of privilege extraction and recognition. However, we prefer to consider the combination of realistic deployment feasibility and detection performance. Therefore, based on the hottest improvement methods of YOLOV5s, we try to make adaptive improvements in three aspects, namely attention mechanism, head network, and backbone network. The experimental results proved that the decoupled head and Slimneck based improvements achieved, respectively, 0.872 and 0.849, 0.538 and 0.479, 87.5% and 89.8% on the mAP0.5, mAP0.5:0.95, and Precision metrics, surpassing the results of the baseline model on these three metrics: 0.705, 0.405 and 83.6%. This result suggests that the adaptively improved model can better meet routine testing needs without significantly increasing the number of parameters. These models perform well on our custom dataset and are also effective on images that are difficult to detect by naked eye. Meanwhile, we find that YOLOV8s, which also has the decoupled head improvement, has the results of 0.743, 0.461, and 87.17% on these three metrics. It proves that under our dataset, it is possible to achieve more advanced results with lower number of model parameters just by adding decoupled head. And according to the results, we also discuss and analyze some improvements that are not adapted to our dataset, which also provides ideas for researchers in similar scenarios: in the booming development of object detection, choosing the suitable model and adapting to combine with other technologies would help to provide solutions to real-world problems.
引用
收藏
页数:23
相关论文
共 50 条
  • [1] Adaptive Dehazing YOLO for Object Detection
    Zhang, Kaiwen
    Yan, Xuefeng
    Wang, Yongzhen
    Qi, Junchen
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2023, PT VII, 2023, 14260 : 14 - 27
  • [2] Tiny YOLO Optimization Oriented Bus Passenger Object Detection
    ZHANG Shuo
    WU Yanxia
    MEN Chaoguang
    LI Xiaosong
    ChineseJournalofElectronics, 2020, 29 (01) : 132 - 138
  • [3] Tiny YOLO Optimization Oriented Bus Passenger Object Detection
    Zhang, Shuo
    Wu, Yanxia
    Men, Chaoguang
    Li, Xiaosong
    CHINESE JOURNAL OF ELECTRONICS, 2020, 29 (01) : 132 - 138
  • [4] Adaptive Tiny Object Detection for Improving Pest Detection
    Huang, Renjie
    He, Yuting
    Xiao, Guoqiang
    Shi, Yangguang
    Zheng, Yongqiang
    2022 26TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2022, : 4544 - 4551
  • [5] Multi Camera Localization Handover Based on YOLO Object Detection Algorithm in Complex Environments
    Wu, Wennan
    Lai, Jizhou
    IEEE ACCESS, 2024, 12 : 15236 - 15250
  • [6] STC-YOLO: Small Object Detection Network for Traffic Signs in Complex Environments
    Lai, Huaqing
    Chen, Liangyan
    Liu, Weihua
    Yan, Zi
    Ye, Sheng
    SENSORS, 2023, 23 (11)
  • [7] YOLO-CFruit: a robust object detection method for Camellia oleifera fruit in complex environments
    Luo, Yuanyin
    Liu, Yang
    Wang, Haorui
    Chen, Haifei
    Liao, Kai
    Li, Lijun
    FRONTIERS IN PLANT SCIENCE, 2024, 15
  • [8] YOLO-GCV: a lightweight algorithm for ship object detection in complex inland waterway environments
    Jin, Yang
    Wang, Ping
    Liu, Shuwang
    Xue, Kai
    Li, Qiuhong
    Wang, Hao
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2025, 22 (01)
  • [9] An Adaptive Sample Assignment Network for Tiny Object Detection
    Dai, Honghao
    Gao, Shanshan
    Huang, Hong
    Mao, Deqian
    Zhang, Chenhao
    Zhou, Yuanfeng
    IEEE TRANSACTIONS ON MULTIMEDIA, 2024, 26 : 2918 - 2931
  • [10] An Adaptive Sample Assignment Network for Tiny Object Detection
    Dai, Honghao
    Gao, Shanshan
    Huang, Hong
    Mao, Deqian
    Zhang, Chenhao
    Zhou, Yuanfeng
    IEEE Transactions on Multimedia, 2024, 26 : 2918 - 2931