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