Optoelectronics Letters, Volume. 20, Issue 6, 372(2024)

Fusion network for small target detection based on YOLO and attention mechanism

Caie XU1...2,3,*, Zhe DONG1, Shengyun ZHONG1, Yijiang CHEN1, Sishun PAN1 and Mingyang and WU1 |Show fewer author(s)
Author Affiliations
  • 1School of Information and Electronic Engineering, Zhejiang University of Science and Technology, Hangzhou 310012, China
  • 2Research Development Department, Hangzhou Xinhe Data Technology Co., Ltd., Hangzhou 311202, China
  • 3College of Mechanical Engineering, Zhejiang University, Hangzhou 310013, China
  • show less

    Target detection is an important task in computer vision research, and such an anomaly detection and the topic of small target detection task is more concerned. However, there are still some problems in this kind of researches, such as small target detection in complex environments is susceptible to background interference and poor detection results. To solve these issues, this study proposes a method which introduces the attention mechanism into the you only look once (YOLO) network. In addition, the amateur-produced mask dataset was created and experiments were conducted. The results showed that the detection effect of the proposed mothed is much better.

    Tools

    Get Citation

    Copy Citation Text

    XU Caie, DONG Zhe, ZHONG Shengyun, CHEN Yijiang, PAN Sishun, and WU Mingyang. Fusion network for small target detection based on YOLO and attention mechanism[J]. Optoelectronics Letters, 2024, 20(6): 372

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Aug. 28, 2023

    Accepted: Nov. 11, 2023

    Published Online: Aug. 23, 2024

    The Author Email: Caie XU (caiexu@163.com)

    DOI:10.1007/s11801-024-3177-3

    Topics