<p>In response to the issues of large target scale variations and complex background environments in ship detection, this paper presents a single-stage ship detection algorithm named DrbLSK, which is an improvement on YOLOv5. First, to enhance the model’s ability to model the context of targets with different scales, a kernel selection module is introduced into the backbone network. By dynamically selecting the receptive field, environmental interference is reduced. Simultaneously, the combination of large-kernel dilated convolution and structural reparameterization is employed to reduce model parameters and enhance the feature expression capability of the backbone network. Next, a decoupled detection head is utilized to alleviate the conflict between classification and regression tasks in the target detection task. Moreover, CIoU is used in the detection head to replace the original loss function, thereby accelerating the convergence of the network. Experimental results show that, on the ABOships dataset, the improved model reduces the number of parameters while increasing accuracy by 2.5% compared to YOLOv5, with a mean average precision of 63.9.</p>

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

Ship detection algorithm based on structural reparameterize dilated large-kernel convolution and spatial selective kernel mechanism

  • Yujing Shi,
  • Haicheng Wang,
  • Shanqiang Li

摘要

In response to the issues of large target scale variations and complex background environments in ship detection, this paper presents a single-stage ship detection algorithm named DrbLSK, which is an improvement on YOLOv5. First, to enhance the model’s ability to model the context of targets with different scales, a kernel selection module is introduced into the backbone network. By dynamically selecting the receptive field, environmental interference is reduced. Simultaneously, the combination of large-kernel dilated convolution and structural reparameterization is employed to reduce model parameters and enhance the feature expression capability of the backbone network. Next, a decoupled detection head is utilized to alleviate the conflict between classification and regression tasks in the target detection task. Moreover, CIoU is used in the detection head to replace the original loss function, thereby accelerating the convergence of the network. Experimental results show that, on the ABOships dataset, the improved model reduces the number of parameters while increasing accuracy by 2.5% compared to YOLOv5, with a mean average precision of 63.9.