<p>Optical remote sensing images have unique advantages in the marine environment, but traditional target detection algorithms have limited performance due to complex background, small target size and dense target distribution. In this study, we propose the YOLOv5–BSD algorithm based on YOLOv5, which optimises the feature fusion network and combines a weighted feature pyramid network and a novel detection head to improve the detection accuracy. The SIoU loss function with angular penalty is introduced to reduce the degree of freedom, and deformable convolution is used to improve the generalization ability. Experiments on the ShipRSImageNet V1.0 data set show that YOLOv5–BSD improves the mAP@0.5, mAP@0.5:0.95, precision and recall by 2.5, 3.7, 2.48 and 2.15 percentage points, respectively, which verifies the effectiveness of the optimization strategy.</p>

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

Yolov5-BSD: a maritime remote sensing object detector based on Yolov5

  • Tianyi Liu,
  • Xiantao Jiang,
  • Qi Cen,
  • Tian Song

摘要

Optical remote sensing images have unique advantages in the marine environment, but traditional target detection algorithms have limited performance due to complex background, small target size and dense target distribution. In this study, we propose the YOLOv5–BSD algorithm based on YOLOv5, which optimises the feature fusion network and combines a weighted feature pyramid network and a novel detection head to improve the detection accuracy. The SIoU loss function with angular penalty is introduced to reduce the degree of freedom, and deformable convolution is used to improve the generalization ability. Experiments on the ShipRSImageNet V1.0 data set show that YOLOv5–BSD improves the mAP@0.5, mAP@0.5:0.95, precision and recall by 2.5, 3.7, 2.48 and 2.15 percentage points, respectively, which verifies the effectiveness of the optimization strategy.