Laser & Optoelectronics Progress, Volume. 60, Issue 10, 1010014(2023)

Marine Fish Detection Algorithm Based on RetinaNet

Yingfeng Zhou, Rongfen Zhang, Yuhong Liu*, and Kuan Li
Author Affiliations
  • College of Big Data and Information Engineering, Guizhou University, Guiyang 550025, Guizhou, China
  • show less
    Figures & Tables(15)
    Structure of improved RetinaNet
    Structure of DenseNet
    Network structure of dense block
    Structure of CBAM
    Structure of CAM
    Structure of SAM
    Structure of PFPN
    Sample diagram of dataset
    Loss value change curve
    Comparison of accuracy of each category
    Comparison of detection results. (a) Detection result of original algorithm; (b) detection result of improved algorithm
    • Table 1. Structure of DenseNet-121

      View table

      Table 1. Structure of DenseNet-121

      LayerSettingOutput size
      Convolution7*7 conv,stride 2112*112
      Pooling3*3 MaxPool,stride 256*56
      Dense block 1[1*1 conv,3*3 conv]*656*56
      Transition layer 11*1 conv,2*2 AvgPool,stride 228*28
      Dense block 2[1*1 conv,3*3 conv]*1228*28
      Transition layer 21*1 conv,2*2 AvgPool,stride 214*14
      Dense block 3[1*1 conv,3*3 conv]*2414*14
      Transition layer 31*1 conv,2*2 AvgPool,stride 27*7
      Dense block 4[1*1 conv,3*3 conv]*167*7
      Classification layer

      7*7 global AvgPool,

      1000D fully-connected,softmax

      1*1
    • Table 2. Experimental environment

      View table

      Table 2. Experimental environment

      CategoryParameter
      CPUIntel(R)Core(TM)i7-7800X
      GPUNvidia GTX 1080Ti

      Operating system

      Development software

      Development framework

      Ubuntu16.04

      PyCharm

      Pytorch

    • Table 3. Precision comparison of different algorithms

      View table

      Table 3. Precision comparison of different algorithms

      AlgorithmPrecision /%mAP /%
      CenterNet84.6584.89
      YOLOV482.1385.77
      Efficientdet86.7287.51
      YOLOV387.5985.79
      SSD84.6786.11
      Ref.[20-85.99
      Ref.[21-89.00
      Proposed92.7892.12
    • Table 4. Ablation experiment

      View table

      Table 4. Ablation experiment

      GroupMethodmAP /%Speed /(ms·image-1Memory /MBModel size /MB
      0RetinaNet87.41894216242.0
      1DenseNet-12188.75463247165.1
      2DenseNet-121+CBAM91.21513358168.5
      3DenseNet-121+PFPN90.53503471167.7
      4DenseNet-121+soft-NMS88.95483201165.5
      5DenseNet-121+CBAM+PFPN91.75543411168.7
      6DenseNet-121+CBAM+soft-NMS91.62533315168.9
      7DenseNet-121+PFPN+soft-NMS90.94523398168.4
      8Proposed92.12563408166.6
    Tools

    Get Citation

    Copy Citation Text

    Yingfeng Zhou, Rongfen Zhang, Yuhong Liu, Kuan Li. Marine Fish Detection Algorithm Based on RetinaNet[J]. Laser & Optoelectronics Progress, 2023, 60(10): 1010014

    Download Citation

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

    Category: Image Processing

    Received: Dec. 27, 2021

    Accepted: Mar. 2, 2022

    Published Online: May. 10, 2023

    The Author Email: Yuhong Liu (1459539967@qq.com)

    DOI:10.3788/LOP213356

    Topics