闵锋,刘朋.改进YOLOv5的SAR图像近海岸舰船目标检测算法研究[J]. 微电子学与计算机,2023,40(4):38-46. doi: 10.19304/J.ISSN1000-7180.2022.0435
引用本文: 闵锋,刘朋.改进YOLOv5的SAR图像近海岸舰船目标检测算法研究[J]. 微电子学与计算机,2023,40(4):38-46. doi: 10.19304/J.ISSN1000-7180.2022.0435
MIN F,LIU P. Research on the detection algorithm of near-coastal ships in SAR images based on improved YOLOv5[J]. Microelectronics & Computer,2023,40(4):38-46. doi: 10.19304/J.ISSN1000-7180.2022.0435
Citation: MIN F,LIU P. Research on the detection algorithm of near-coastal ships in SAR images based on improved YOLOv5[J]. Microelectronics & Computer,2023,40(4):38-46. doi: 10.19304/J.ISSN1000-7180.2022.0435

改进YOLOv5的SAR图像近海岸舰船目标检测算法研究

Research on the detection algorithm of near-coastal ships in SAR images based on improved YOLOv5

  • 摘要: SAR图像舰船目标检测时,因近海岸港口存在着复杂背景的问题,以至于重叠舰船目标无法被准确提取特征信息,造成近海岸的舰船目标出现漏检、误检的情况. 针对以上问题,提出一种复杂场景下的SAR图像舰船检测算法,该算法基于YOLOv5进行改进,采用SPPF结构加强提取特征信息,并融合原YOLOv5的SPP结构提取的特征信息,这种多级金字塔模块并列融合的方式能有效的检测多尺度舰船目标,使特征信息更好的表达;然后将原模型中的GIOU改进为CIOU,使其可以准确的回归出预测框的位置;最终为了更合理的筛选高于阈值的预测框,改进NMS(Non-Maximum-Suppression),采用Soft-NMS方法去惩罚衰减高于阈值的边框得分,合理的去除预测框. 试验结果表明,该文改进的模型相比于原模型在SSDD、SAR-Ship-Dataset数据集上的mAP(mean Average Precision)提高了5.15%和5.06%,改进模型能有效检测近海岸中复杂背景下的SAR图像舰船目标.

     

    Abstract: During the detection of ship targets in SAR images, due to the complex background problems in near-coastal ports, the feature information of overlapping ship targets cannot be accurately extracted, resulting in missed detection and false detection of near-coastal ship targets. Aiming at the above problems, this paper proposes a SAR image ship detection algorithm in complex scenes. The algorithm is improved based on YOLOv5, uses the SPPF structure to enhance the extraction of feature information, and fuses the feature information extracted by the SPP structure of the original YOLOv5. This multi-level pyramid The method of parallel fusion of modules can effectively detect multi-scale ship targets, so that the feature information can be better expressed; then the GIOU in the original model is improved to CIOU, so that it can accurately return to the position of the prediction frame; finally, in order to be more reasonable The prediction box above the threshold is screened, NMS (Non-Maximum-Suppression) is improved, and the Soft-NMS method is used to penalize and decay the box score higher than the threshold, and reasonably remove the prediction box. The experimental results show that the mAP (mean Average Precision) of the improved model on SSDD and SAR-Ship-Dataset data sets is improved by 5.15% and 5.06% compared with the original model, and the improved model can effectively detect the complex background in the near coast. SAR image below the ship target.

     

/

返回文章
返回