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 条
  • [21] Image-Adaptive YOLO for Object Detection in Adverse Weather Conditions
    Liu, Wenyu
    Ren, Gaofeng
    Yu, Runsheng
    Guo, Shi
    Zhu, Jianke
    Zhang, Lei
    THIRTY-SIXTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FOURTH CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE / THE TWELVETH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2022, : 1792 - 1800
  • [22] Fog-Aware Adaptive YOLO for Object Detection in Adverse Weather
    Abbasi, Hasan
    Amini, Marzieh
    Yu, F. Richard
    2023 IEEE SENSORS APPLICATIONS SYMPOSIUM, SAS, 2023,
  • [23] STUDY ON FEATURE LAYER OF ADAPTIVE SELECTION PYRAMID FOR SMALL OBJECT DETECTION IN COMPLEX ENVIRONMENTS
    Yang, Xueying
    Zhang, Zhiyong
    Qiu, Changzhen
    Wang, Luping
    FRESENIUS ENVIRONMENTAL BULLETIN, 2021, 30 (01): : 474 - 483
  • [24] MULTISCALE DOMAIN ADAPTIVE YOLO FOR CROSS-DOMAIN OBJECT DETECTION
    Hnewa, Mazin
    Radha, Hayder
    2021 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2021, : 3323 - 3327
  • [25] Adaptive Dynamic Label Assignment for Tiny Object Detection in Aerial Images
    Ge, Lihui
    Wang, Guanqun
    Zhang, Tong
    Zhuang, Yin
    Chen, He
    Dong, Hao
    Chen, Liang
    IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 6201 - 6214
  • [26] YOLO v3-Tiny: Object Detection and Recognition using one stage improved model
    Adarsh, Pranav
    Rathi, Pratibha
    Kumar, Manoj
    2020 6TH INTERNATIONAL CONFERENCE ON ADVANCED COMPUTING AND COMMUNICATION SYSTEMS (ICACCS), 2020, : 687 - 694
  • [27] YOLO with adaptive frame control for real-time object detection applications
    Lee, Jeonghun
    Hwang, Kwang-il
    MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (25) : 36375 - 36396
  • [28] YOLO with adaptive frame control for real-time object detection applications
    Jeonghun Lee
    Kwang-il Hwang
    Multimedia Tools and Applications, 2022, 81 : 36375 - 36396
  • [29] SSDA-YOLO: Semi-supervised domain adaptive YOLO for cross-domain object detection
    Zhou, Huayi
    Jiang, Fei
    Lu, Hongtao
    COMPUTER VISION AND IMAGE UNDERSTANDING, 2023, 229
  • [30] Tinier-YOLO: A Real-Time Object Detection Method for Constrained Environments
    Fang, Wei
    Wang, Lin
    Ren, Peiming
    IEEE ACCESS, 2020, 8 : 1935 - 1944