Suppr超能文献

用于水面目标检测的改进YOLOv8算法

Improved YOLOv8 Algorithm for Water Surface Object Detection.

作者信息

Wang Jie, Zhao Hong

机构信息

Key Laboratory of Advanced Manufacturing and Automation Technology, Education Department of Guangxi Zhuang Autonomous Region, Guilin University of Technology, Guilin 541006, China.

College of Mechanical and Control Engineering, Guilin University of Technology, Guilin 541006, China.

出版信息

Sensors (Basel). 2024 Aug 5;24(15):5059. doi: 10.3390/s24155059.

Abstract

To address the issues of decreased detection accuracy, false detections, and missed detections caused by scale differences between near and distant targets and environmental factors (such as lighting and water waves) in surface target detection tasks for uncrewed vessels, the YOLOv8-MSS algorithm is proposed to be used to optimize the detection of water surface targets. By adding a small target detection head, the model becomes more sensitive and accurate in recognizing small targets. To reduce noise interference caused by complex water surface environments during the downsampling process in the backbone network, C2f_MLCA is used to enhance the robustness and stability of the model. The lightweight model SENetV2 is employed in the neck component to improve the model's performance in detecting small targets and its anti-interference capability. The SIoU loss function enhances detection accuracy and bounding box regression precision through shape awareness and geometric information integration. Experiments on the publicly available dataset FloW-Img show that the improved algorithm achieves an mAP@0.5 of 87.9% and an mAP@0.5:0.95 of 47.6%, which are improvements of 5% and 2.6%, respectively, compared to the original model.

摘要

为了解决无人船水面目标检测任务中,由于近远目标尺度差异以及环境因素(如光照和水波)导致的检测精度下降、误检和漏检问题,提出使用YOLOv8-MSS算法来优化水面目标检测。通过添加一个小目标检测头,模型在识别小目标时变得更加灵敏和准确。为了减少骨干网络下采样过程中复杂水面环境引起的噪声干扰,采用C2f_MLCA来增强模型的鲁棒性和稳定性。在颈部组件中使用轻量级模型SENetV2,以提高模型检测小目标的性能及其抗干扰能力。SIoU损失函数通过形状感知和几何信息整合提高检测精度和边界框回归精度。在公开可用数据集FloW-Img上的实验表明,改进后的算法mAP@0.5达到87.9%,mAP@0.5:0.95达到47.6%,与原始模型相比,分别提高了5%和2.6%。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/d07f/11314716/0e4c8ff5b99d/sensors-24-05059-g001.jpg

文献检索

告别复杂PubMed语法,用中文像聊天一样搜索,搜遍4000万医学文献。AI智能推荐,让科研检索更轻松。

立即免费搜索

文件翻译

保留排版,准确专业,支持PDF/Word/PPT等文件格式,支持 12+语言互译。

免费翻译文档

深度研究

AI帮你快速写综述,25分钟生成高质量综述,智能提取关键信息,辅助科研写作。

立即免费体验