• Electronics Optics & Control
  • Vol. 31, Issue 9, 38 (2024)
ZHANG Lin, BO Jingdong, GONG Ruikun, and CUI Chuanjin
Author Affiliations
  • [in Chinese]
  • show less
    DOI: 10.3969/j.issn.1671-637x.2024.09.007 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.