Electronics Optics & Control, Volume. 31, Issue 9, 38(2024)

An Infrared Ship Target Detection Algorithm Based on Improved YOLOv5s

ZHANG Lin... BO Jingdong, GONG Ruikun and CUI Chuanjin |Show fewer author(s)
Author Affiliations
  • [in Chinese]
  • show less

    A lightweight infrared ship detection algorithm CBYOLOv5 combined with Knowledge Distillation (KD) is proposed to solve the problem of large amount of parameters and computation in traditional ship detection algorithms. Lightweight network Ghost module is introduced in YOLOv5s backbone network to realize lightweight detection network. A new neck structure of Asymptotic Feature Pyramid Network (AFPN) is introducedwhich can avoid the large semantic gap of nonadjacent level by fusing two adjacent lowlevel features and gradually fusing to higherlevel features. The VFL function is used to improve the imbalance of positive and negative samples in infrared ship target detection tasksso as to improve the overall performance of the model. FinallyKD is adopted to transfer the “knowledge” in the network of teachers with strong learning ability into the improved network model to improve the accuracy of classification and localization. Experimental results show that in infrared ship datasetin comparison with original algorithm YOLOv5sparameter amount is reduced by 38%and mAP is increased by 3.9 percentage pointswhile the model weight file is only 8.96×106which proves the proposed algorithm is effective and has certain practical value.

    Tools

    Get Citation

    Copy Citation Text

    ZHANG Lin, BO Jingdong, GONG Ruikun, CUI Chuanjin. An Infrared Ship Target Detection Algorithm Based on Improved YOLOv5s[J]. Electronics Optics & Control, 2024, 31(9): 38

    Download Citation

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

    Received: Sep. 11, 2023

    Accepted: --

    Published Online: Oct. 22, 2024

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2024.09.007

    Topics