• Opto-Electronic Engineering
  • Vol. 51, Issue 1, 230284-1 (2024)
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
    DOI: 10.12086/oee.2024.230284 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.
    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