Opto-Electronic Engineering, Volume. 51, Issue 1, 230284-1(2024)

Lightweight YOLOv5 sonar image object detection algorithm and implementation based on ZYNQ

Dongdong Zhao1, Dunhan Xie1, Peng Chen1、*, Ronghua Liang1, Yi Shen1, and Xinxin Guo2
Author Affiliations
  • 1College of Computer Science and Technology, Zhejiang University of Technology, Hangzhou, Zhejiang 310023, China
  • 2Institute of Deep-sea Science and Engineering, Chinese Academy of Sciences, Sanya, Hainan 572000, China
  • show less

    To address the problems of blurring and insufficient sample size in sonar images, an improved sonar image target detection algorithm is proposed based on YOLOv5. The algorithm uses geometric filtering, vertical flipping, and other methods to enhance the sonar image dataset. The fusion attention mechanism module is added to make the algorithm better focus on the features of small targets in sonar images. At the same time, in response to the problem that most target detection algorithms currently run on the cloud and cannot achieve real-time sonar image detection, this paper uses lightweight network replacement and NCNN edge porting technology. It adopts the GSConv module in the neck network to successfully transplant the algorithm to the ZYNQ platform, realizing real-time detection of sonar images on the embedded end. After experiments, the algorithm proposed in this paper reduced the parameter quantity by 56%, increasing map50 and map50-95 by 2.2% and 2.5%, respectively. The algorithm’s performance has significantly improved, proving that the method proposed has certain feasibility and effectiveness in lightweight sonar image target detection tasks.

    Tools

    Get Citation

    Copy Citation Text

    Dongdong Zhao, Dunhan Xie, Peng Chen, Ronghua Liang, Yi Shen, Xinxin Guo. Lightweight YOLOv5 sonar image object detection algorithm and implementation based on ZYNQ[J]. Opto-Electronic Engineering, 2024, 51(1): 230284-1

    Download Citation

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

    Category: Research Articles

    Received: Nov. 21, 2023

    Accepted: Jan. 10, 2024

    Published Online: Apr. 19, 2024

    The Author Email: Chen Peng (陈朋)

    DOI:10.12086/oee.2024.230284

    Topics