一种内河船舶目标检测方法及系统

本发明提供一种内河船舶目标检测方法及系统，涉及船舶图像识别技术领域。该方法步骤包括：建立内河船舶目标数据集；对内河船舶目标数据集进行数据增强处理；采用BiFPN模块替换YOLOv8模型中的PANet模块；在颈部网络中引入SimAm注意力模块；采用RepConv卷积对检测头进行改进；引入Shape‑IoU代替CIoU作为YOLOv8模型的边界框损失计算函数；得到改进的YOLOv8模型后，利用内河船舶目标数据集训练后得到内河船舶目标检测模型，实现待测图像的目标检测。本发明能够在保证实时性的情况下，有效平衡检测精度和模型复杂度，在小目标检测和复杂环境下具有明显优势。