首页    期刊浏览 2024年07月08日 星期一
登录注册

文章基本信息

  • 标题:Underwater Target Detection Algorithm Based on Improved YOLOv5
  • 本地全文:下载
  • 作者:Fei Lei ; Feifei Tang ; Shuhan Li
  • 期刊名称:Journal of Marine Science and Engineering
  • 电子版ISSN:2077-1312
  • 出版年度:2022
  • 卷号:10
  • 期号:1
  • 页码:310
  • DOI:10.3390/jmse10030310
  • 语种:English
  • 出版社:MDPI AG
  • 摘要:Underwater target detection plays an important role in ocean exploration, to which the improvement of relevant technology is of much practical significance. Although existing target detection algorithms have achieved excellent performance on land, they often fail to achieve satisfactory outcome of detection when in the underwater environment. In this paper, one of the most advanced target detection algorithms, YOLOv5 (You Only Look Once), was first applied in the underwater environment before being improved by combining it with some methods characteristic of the underwater environment. To be specific, the Swin Transformer was treated as the basic backbone network of YOLOv5, which makes the network suitable for those underwater images with blurred targets. It is possible for the network to focus on fusing the relatively important resolution features by improving the method of path aggregation network (PANet) for multi-scale feature fusion. The confidence loss function was improved on the basis of different detection layers, with the network biased to learn high-quality positive anchor boxes and make the network more capable of detecting the target. As suggested by the experimental results, the improved network model is effective in detecting underwater targets, with the mean average precision (mAP) reaching 87.2%, which makes it advantageous over general target detection models and fit for use in the complex underwater environment.
国家哲学社会科学文献中心版权所有