面向无人艇自主航行下的水面检测模型

Water Surface Detection Model for Autonomous Navigation of Unmanned Vessels

  • 摘要: 针对无人艇在复杂多场景水面环境下执行水面清理等任务时会存在目标漏检、误检等问题,提出一种基于YOLOv8s改进的水面目标检测模型。首先,对小目标检测层进行重构,引入低层特征细节信息,旨在降低模型参数并提高模型对小目标的感知能力; 其次,引入部分卷积PConv代替传统Conv并构建特征提取模块P-C2f,旨在减少冗余特征和计算,进一步压缩模型大小;接着,使用重参数化泛化特征金字塔网络RepGFPN来融合特征,旨在加强低层细节信息和高层语义信息的交互融合,提高模型对多尺度目标的检测能力;最后,使用迁移学习对模型进行微调,进一步提高检测性能。在WSODD数据集上进行测试,改进模型较原模型在参数量下降近67.5%的同时,召回率R提升了4%,mAP@0.5提升了2.1%,达到了81.4%,且与其他主流检测模型相比有明显优势,能帮助无人艇更好地执行水面检测任务。

     

    Abstract: To address the problems of missed and false detections of surface targets by unmanned vessels operating in complex multi-scenario water environments, a surface object detection model based on improved YOLOv8s is proposed. First, the small-target detection layer is reconstructed by introducing low-level feature details to reduce the number of model parameters and improve the model's perception of small targets. Second, partial convolution (PConv) is introduced to replace the traditional Conv and construct the feature extraction module P-C2f, aiming to reduce redundant features and computation, which further compress the model size. Subsequently, the reparameterized generalized feature pyramid network (RepGFPN) is used to fuse features, aiming to enhance the full interaction and fusion of low-level detail information and high-level semantic information, thereby improving the multiscale target detection ability of the model. Finally, transfer learning is used to fine-tune the model and further improve detection performance. When tested on the WSODD dataset, the reference number of the improved model decreased by nearly 67.5% compared to that of the original model, whereas recall rate R increased by 4%, and mAP@0.5 increased by 2.1%, reaching 81.4%. Compared with other mainstream detection models, the improved model has obvious advantages and can help unmanned vehicles perform better surface detection tasks.

     

/

返回文章
返回