Back to articles
Regular Article
Volume: 68 | Article ID: 030502
Image
Underwater Object Detection Algorithm based on Lightweight Structure Optimization
  DOI :  10.2352/J.ImagingSci.Technol.2024.68.3.030502  Published OnlineMay 2024
Abstract
Abstract

This paper proposes an underwater object detection algorithm based on lightweight structure optimization to address the low detection accuracy and difficult deployment in underwater robot dynamic inspection caused by low light, blurriness, and low contrast. The algorithm builds upon YOLOv7 by incorporating the attention mechanism of the convolutional module into the backbone network to enhance feature extraction in low light and blurred environments. Furthermore, the feature fusion enhancement module is optimized to control the shortest and longest gradient paths for fusion, improving the feature fusion ability while reducing network complexity and size. The output module of the network is also optimized to improve convergence speed and detection accuracy for underwater fuzzy objects. Experimental verification using real low-light underwater images demonstrates that the optimized network improves the object detection accuracy (mAP) by 11.7%, the detection rate by 2.9%, and the recall rate by 15.7%. Moreover, it reduces the model size by 20.2 MB with a compression ratio of 27%, making it more suitable for deployment in underwater robot applications.

Subject Areas :
Views 12
Downloads 5
 articleview.views 12
 articleview.downloads 5
  Cite this article 

Liang Chen, Jin Zhao, Junwei Yang, Huihui Guo, Shaowu Zhou, "Underwater Object Detection Algorithm based on Lightweight Structure Optimizationin Journal of Imaging Science and Technology,  2024,  pp 1 - 12,  https://doi.org/10.2352/J.ImagingSci.Technol.2024.68.3.030502

 Copy citation
  Copyright statement 
Copyright © Society for Imaging Science and Technology 2024
  Article timeline 
  • received September 2023
  • accepted December 2023
  • PublishedMay 2024

Preprint submitted to:
  Login or subscribe to view the content