Coastal Vessel Target Detection Model Based on Improved YOLOv7
摘要
To address low detection accuracy in near-coastal vessel target detection under complex conditions, a novel near-coastal vessel detection model based on an improved YOLOv7 architecture is proposed in this paper. The attention mechanism Coordinate Attention is used to improve channel attention weight and enhance a network’s ability to extract small target features. In the enhanced feature extraction network, the lightweight convolution algorithm Grouped Spatial Convolution is used to replace MPConv to reduce model calculation costs. EIoU Loss is used to replace the regression frame loss function in YOLOv7 to reduce the probability of missed and false detection. The performance of the improved model was verified using an enhanced dataset obtained through rainy and foggy weather simulation. Experiments were conducted on the datasets before and after the enhancement. The improved model achieved a mean average precision (mAP) of 97.45% on the original dataset, and the number of parameters was reduced by 2%. On the enhanced dataset, the mAP of the improved model reached 88.08%. Compared with seven target detection models, such as Faster R-CNN, YOLOv3, YOLOv4, YOLOv5, YOLOv7, YOLOv8-n, and YOLOv8-s, the improved model can effectively reduce the missed and false detection rates and improve target detection accuracy. The improved model not only accurately detects vessels in complex weather environments but also outperforms other methods on original and enhanced SeaShip datasets. This finding shows that the improved model can achieve near-coastal vessel target detection in multiple environments, laying the foundation for vessel path planning and automatic obstacle avoidance.