Journal of Marine Science and Engineering (Aug 2024)

Real-Time Underwater Fish Detection and Recognition Based on CBAM-YOLO Network with Lightweight Design

  • Zheping Yan,
  • Lichao Hao,
  • Jianmin Yang,
  • Jiajia Zhou

DOI
https://doi.org/10.3390/jmse12081302
Journal volume & issue
Vol. 12, no. 8
p. 1302

Abstract

Read online

More and more underwater robots are deployed to investigate marine biodiversity autonomously, and tools are needed by underwater robots to discover and acknowledge marine life. This paper has proposed a convolutional neural network-based method for intelligent fish detection and recognition with a dataset used for training and testing generated and augmented from an open-source Fish Database regarding 6 different types. Firstly, to improve image quality, a hybrid image enhancement algorithm is used to preprocess underwater images with a weighted fusion strategy of multiple traditional methodologies and comparisons have been made to prove the effectiveness according to various indexes. Secondly, to increase detection and recognition accuracy, different attention modules are integrated into the YOLOv5m network structure and the convolutional block attention module(CBAM) has outperformed other modules in recall rate and mAP while maintaining the capability of real-time processing. Lastly, to meet real-time requirements, lightweight adjustments have been made to CBAM-YOLOv5m with the GSConv module and C3Ghost module and a nearly 25% reduction in network parameters and a 20% reduction in computational consumption are obtained. Besides, the lightweight network has realized better accuracy than YOLOv5m. In conclusion, the method proposed in this paper is effective in real-time fish detection and recognition with practical application prospects.

Keywords