一种基于YOLOv5的船舶检测方法

本发明提供一种基于YOLOv5的船舶检测方法，包括：获取带有目标图像的船舶检测数据集；构建改进的YOLOv5检测模型，以YOLOv5为基础模型，将主干网络替换成MobileNetv3网络，在颈部网络中加入SimAM模块，将原特征金字塔网络结构替换为BiFPN网络结构；将CIoU损失函数替换为EIoU损失函数；通过船舶检测数据集对改进的YOLOv5检测模型进行训练；利用评价指标对训练好的检测模型进行评估，得到目标检测模型；将含有待检测目标的图像作为输入，通过目标检测模型对图像中的待检测目标进行检测；本发明算法复杂度低、识别精度高、误检和漏检率低、能够应对水面复杂环境的干扰。