一种基于改进YOLOv11n的船舶小目标检测算法:WTDS-YOLO

WTDS-YOLO: A Small Object Detection Algorithm for Ships Based on an Improved YOLOv11n

  • 摘要: 【目的】针对船舶检测在小目标密集、背景干扰严重的滨海环境下易出现漏检和误检的问题。【方法】提出一种基于WTDS-YOLO模型的船舶目标检测方法,以YOLOv11n为基础模型,使用轻量化小波变换卷积WTConv与可分离卷积相结合,增强模型抗噪声能力。其次,嵌入动态多尺度自适应特征调制模块DyMSAFM,在保持速度的前提下提升小目标检测能力。最后,通过ShapeIoU Loss进一步提升模型推理速度。【结果】结果表明,改进模型的平均精度mAP50、召回率Recall以及F1score分别达到了87.6%、77.8%、83.0%,与YOLOv11n原模型相比分别提升了5.4%、5.6%、6.0%。在显著提升精度的同时,达到101.56 FPS的推理速度。【结论】所提算法精度高、速度快,适用于实时海上监测和港口管理等场景。

     

    Abstract: Objectives To address the issues of missed detections and false positives in ship detection within coastal environments characterized by dense small targets and severe background interference. Methods A ship target detection method based on the WTDS-YOLO model is proposed. Using YOLOv11n as the base model, it combines lightweight wavelet transform convolutions (WTConv) with separable convolutions to enhance the model's noise resistance. Second, a Dynamic Multi-Scale Adaptive Feature Modulation (DyMSAFM) module is embedded to enhance small target detection capability while maintaining processing speed. Finally, ShapeIoU Loss is employed to further accelerate model inference. Results The model achieves an average precision (mAP50) of 87.6%, recall of 77.8%, and F1 score of 83.0%, representing improvements of 5.4%, 5.6%, and 6.0% respectively over the original YOLOv11n model. While significantly enhancing accuracy, it achieves an inference speed of 101.56 FPS. Conclusions The proposed algorithm delivers high accuracy and fast processing speed, making it suitable for real-time maritime monitoring and port management scenarios.

     

/

返回文章
返回