Electronics Optics & Control, Volume. 31, Issue 9, 38(2024)
An Infrared Ship Target Detection Algorithm Based on Improved YOLOv5s
A lightweight infrared ship detection algorithm CBYOLOv5 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 introducedwhich can avoid the large semantic gap of nonadjacent level by fusing two adjacent lowlevel features and gradually fusing to higherlevel features. The VFL function is used to improve the imbalance of positive and negative samples in infrared ship target detection tasksso as to improve the overall performance of the model. FinallyKD 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 datasetin comparison with original algorithm YOLOv5sparameter amount is reduced by 38%and mAP is increased by 3.9 percentage pointswhile the model weight file is only 8.96×106which proves the proposed algorithm is effective and has certain practical value.
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
Received: Sep. 11, 2023
Accepted: --
Published Online: Oct. 22, 2024
The Author Email: