错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Ship Target Intelligent Detection Algorithm Based on Improved YOLOv5s

  • Mengchen Dong,
  • Jian Yang,
  • Fangyuan Xie,
  • Yunzhi Ling,
  • Weizhong Yu

摘要

Currently, the deep learning-based synthetic aperture radar (SAR) image target detection methods have limited detection ability on small targets, and their accuracy needs to be improved either. In order to improve the method’s recognition ability and accuracy on small targets, this paper introduced the coordinate attention mechanism to obtain the long-range dependency relationship based on YOLOv5s model, and used SIoU loss function to improve the detection speed and accuracy of the model. The optimized YOLOv5 algorithm are significantly improved the performance and calculation speed of the detector. The experimental results on the High-Resolution SAR Image Dataset (HRSID), indicated that the accuracy and recall rate of the proposed method are increased by 2.566 \(\%\) and 0.871 \(\%\) , respectively. At the same time, mAP0.5 is increased by 0.579%, mAP0.5-0.95 is increased by 2.737 \(\%\) , the training time is significantly reduced, and the detection performance is greatly improved.