• Infrared and Laser Engineering
  • Vol. 52, Issue 10, 20230006 (2023)
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
    DOI: 10.3788/IRLA20230006 Cite this Article
    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 Copy Citation Text show less
    Structure diagram of YOLOv5
    Fig. 1. Structure diagram of YOLOv5
    The flow chart of target detection
    Fig. 2. The flow chart of target detection
    Normalized border length and width distribution
    Fig. 3. Normalized border length and width distribution
    Distribution of clustering results
    Fig. 4. Distribution of clustering results
    mAP curve of clustering algorithm before and after improvement
    Fig. 5. mAP curve of clustering algorithm before and after improvement
    Schematic diagram of intersection over union
    Fig. 6. Schematic diagram of intersection over union
    Target box and prediction border
    Fig. 7. Target box and prediction border
    Comparison of visual legend 1 of multiple loss functions
    Fig. 8. Comparison of visual legend 1 of multiple loss functions
    Comparison of visual legend 2 of multiple loss functions
    Fig. 9. Comparison of visual legend 2 of multiple loss functions
    Loss curve of the MIoU
    Fig. 10. Loss curve of the MIoU
    Comparison chart of detection results of different algorithms
    Fig. 11. Comparison chart of detection results of different algorithms
    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 1. Clustering results
    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 2. Detection accuracy of different detection algorithms
    MethodK-median++MIoU
    a××
    d×
    c×
    Proposed method
    Table 3. Comparison of ablation tests of various improved methods
    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%
    Table 4. Ablation test results
    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