• Laser & Optoelectronics Progress
  • Vol. 60, Issue 2, 0215001 (2023)
Rongrong Wang1 and Zhongyun Jiang2、*
Author Affiliations
  • 1College of Information, Shanghai Ocean University, Shanghai 201306, China
  • 2College of Information Technology, Shanghai Jian Qiao University, Shanghai 201306, China
  • show less
    DOI: 10.3788/LOP212230 Cite this Article Set citation alerts
    Rongrong Wang, Zhongyun Jiang. Underwater Object Detection Algorithm Based on Improved CenterNet[J]. Laser & Optoelectronics Progress, 2023, 60(2): 0215001 Copy Citation Text show less

    Abstract

    Aiming at the problems of conventional detectors in detecting underwater objects, such as difficulty in feature extraction and missing detection of objects, an improved CenterNet underwater object detection method is proposed. First, a high resolution human posture estimation network HRNet is used to replace the Hourglass-104 backbone network in CenterNet model to reduce the amount of parameters and improve the speed of network reasoning; then, the bottleneck attention module is introduced to enhance the features in the spatial and channel dimensions, and improve the detection accuracy; finally, a feature fusion module is constructed to integrate the rich semantic information and spatial location information in the network, the fused features are processed by receptive field block to further improve the multi-scale object detection ability of the network. A comparison experiment is carried out on the URPU underwater object detection dataset. Compared with CenterNet network, the detection accuracy of the proposed algorithm can reach 77.4%, increased by 1.5 percentage points, the detection speed is 7 frame/s, increased by 35.6%, the amount of parameters is 30.4 MB, compressed by 84.1%. Compared with the mainstream object detection algorithm, this algorithm also has higher detection accuracy, which has higher advantages in underwater object detection.
    Rongrong Wang, Zhongyun Jiang. Underwater Object Detection Algorithm Based on Improved CenterNet[J]. Laser & Optoelectronics Progress, 2023, 60(2): 0215001
    Download Citation