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.
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%。