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 条
  • [31] Lemon-YOLO: An efficient object detection method for lemons in the natural environment
    Li, Guojin
    Huang, Xiaojie
    Ai, Jiaoyan
    Yi, Zeren
    Xie, Wei
    IET IMAGE PROCESSING, 2021, 15 (09) : 1998 - 2009
  • [32] YOLO-CRD: A Lightweight Model for the Detection of Rice Diseases in Natural Environments
    Zhang, Rui
    Liu, Tonghai
    Liu, Wenzheng
    Yuan, Chaungchuang
    Seng, Xiaoyue
    Guo, Tiantian
    Wang, Xue
    PHYTON-INTERNATIONAL JOURNAL OF EXPERIMENTAL BOTANY, 2024, 93 (06) : 1275 - 1296
  • [33] GSBF-YOLO: a lightweight model for tomato ripeness detection in natural environments
    Hao, Fengqi
    Zhang, Zuyao
    Ma, Dexin
    Kong, Hoiio
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2025, 22 (01)
  • [34] DNE-YOLO: A method for apple fruit detection in Diverse Natural Environments
    Wu, Haitao
    Mo, Xiaotian
    Wen, Sijian
    Wu, Kanglei
    Ye, Yu
    Wang, Yongmei
    Zhang, Youhua
    JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2024, 36 (09)
  • [35] Research on Object Detection Method Based on FF-YOLO for Complex Scenes
    Chen Baoyuan
    Liu Yitong
    Sun Kun
    IEEE ACCESS, 2021, 9 : 127950 - 127960
  • [36] HRYNet: A Highly Robust YOLO Network for Complex Road Traffic Object Detection
    Tang, Lindong
    Yun, Lijun
    Chen, Zaiqing
    Cheng, Feiyan
    SENSORS, 2024, 24 (02)
  • [37] MRD-YOLO: A Multispectral Object Detection Algorithm for Complex Road Scenes
    Sun, Chaoyue
    Chen, Yajun
    Qiu, Xiaoyang
    Li, Rongzhen
    You, Longxiang
    SENSORS, 2024, 24 (10)
  • [38] Adaptive Downsampling and Scale Enhanced Detection Head for Tiny Object Detection in Remote Sensing Image
    Zhang, Yunzuo
    Liu, Ting
    Zhen, Jiawen
    Kang, Yaoxing
    Cheng, Yu
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2025, 22
  • [39] Object Detection in Shelf Images with YOLO
    Melek, Ceren Gulra
    Sonmez, Elena Battini
    Albayrak, Songul
    PROCEEDINGS OF 18TH INTERNATIONAL CONFERENCE ON SMART TECHNOLOGIES (IEEE EUROCON 2019), 2019,
  • [40] A optimized YOLO method for object detection
    Liang Tianjiao
    Bao Hong
    2020 16TH INTERNATIONAL CONFERENCE ON COMPUTATIONAL INTELLIGENCE AND SECURITY (CIS 2020), 2020, : 30 - 34