Infrared and Laser Engineering, Volume. 52, Issue 10, 20230006(2023)

Infrared ship target detection algorithm based on YOLOv5

Fen Liu1, Jie Sun1, Shuai Zhang1, Hongqiang Sang1, and Xiujun Sun1,2、*
Author Affiliations
  • 1School of Mechanical Engineering, Tiangong University, Tianjin 300387, China
  • 2Key Laboratory of Physical Oceanography, MOE, Ocean University of China, Qingdao 266100, China
  • show less
    Figures & Tables(15)
    Structure diagram of YOLOv5
    The flow chart of target detection
    Normalized border length and width distribution
    Distribution of clustering results
    mAP curve of clustering algorithm before and after improvement
    Schematic diagram of intersection over union
    Target box and prediction border
    Comparison of visual legend 1 of multiple loss functions
    Comparison of visual legend 2 of multiple loss functions
    Loss curve of the MIoU
    Comparison chart of detection results of different algorithms
    • Table 1. Clustering results

      View table
      View in Article

      Table 1. Clustering results

      K-meansK-median++
      [10, 13][16, 11]
      [16, 30][42, 17]
      [33, 23][22, 44]
      [30, 61][36, 81]
      [62, 45][100, 29]
      [59, 119][170, 55]
      [116, 90][92, 147]
      [156, 198][160, 221]
      [373, 326][241, 109]
    • Table 2. Detection accuracy of different detection algorithms

      View table
      View in Article

      Table 2. Detection accuracy of different detection algorithms

      AlgorithmsmAP@0.5AP
      LinerBulkWarshipsailboatCanoeContainerFishing
      YOLOv5-SE0.934%0.936%0.985%0.992%0.920%0.889%0.972%0.841%
      CBAM-Yolov50.930%0.940%0.986%0.985%0.916%0.873%0.959%0.844%
      YOLOv50.928%0.939%0.983%0.989%0.915%0.876%0.952%0.839%
      Proposed algorithms0.939%0.944%0.986%0.992%0.916%0.875%0.973%0.889%
    • Table 3. Comparison of ablation tests of various improved methods

      View table
      View in Article

      Table 3. Comparison of ablation tests of various improved methods

      MethodK-median++MIoU
      a××
      d×
      c×
      Proposed method
    • Table 4. Ablation test results

      View table
      View in Article

      Table 4. Ablation test results

      MethodmAP@0.5AP
      LinerBulkWarshipSailboatCanoeContainerFishing
      a92.8%93.9%98.3%98.9%91.5%87.6%95.2%83.9%
      b93.5%93.6%98.2%99.0%90.7%87.9%96.3%88.7%
      c93.0%94.4%98.6%98.5%91.6%87.3%95.9%84.4%
      Proposed method93.9%94.4%98.6%99.2%91.6%87.5%97.3%88.9%
    Tools

    Get Citation

    Copy Citation Text

    Fen Liu, Jie Sun, Shuai Zhang, Hongqiang Sang, Xiujun Sun. Infrared ship target detection algorithm based on YOLOv5[J]. Infrared and Laser Engineering, 2023, 52(10): 20230006

    Download Citation

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

    Category: Image processing

    Received: Jan. 5, 2023

    Accepted: --

    Published Online: Nov. 21, 2023

    The Author Email: Sun Xiujun (.sxj@ouc.edu.cn)

    DOI:10.3788/IRLA20230006

    Topics