首页|期刊导航|交通信息与安全|基于PEW-YOLOv8的内河船舶目标检测方法

基于PEW-YOLOv8的内河船舶目标检测方法OA北大核心

A Method for Inland Vessel Object Detection Based on PEW-YOLOv8

中文摘要英文摘要

内河船舶目标检测中,众多检测对象属于小目标范畴,其在图像中的像素占比有限,且由于水域环境干扰等问题,导致检测精度不足,误检、漏检现象频发,为此研究了PEW-YOLOv8(YOLOv8+P2检测层+Effi-cientNetV2+WIoUinner)目标检测算法.新增160×160分辨率的P2浅层次小目标检测层,通过32维特征空间重构实现多尺度特征的动态权重分配,设计高低层特征的双向交互机制,增强对小型船舶目标的特征提取能力;为应对多层次目标检测头导致的模型训练参数量增加的难题,采用改进的EfficientNetV2高效架构优化策略,引入Stems模块采用GELU激活函数避免梯度爆炸和训练不稳定,训练阶段保留扩展4倍的通道数,简化卷积结构显著加速训练过程,同时保证模型训练质量;设计动态非单调聚焦机制的WIoUinner损失函数,构建具有一定尺度差异的辅助预测框,加速边界框收敛速度,使模型在预测框与真实框重合良好时更注重中心点之间的距离,减轻几何度量的惩罚,从而提升模型的泛化能力.通过融合公开的Seaships数据集与自建数据集形成的数据集进行算法与实验验证,结果表明:同YOLOv10相比,PEW-YOLOv8平均检测精度达到94.8%,提升了3%,计算复杂度显著降低,FLOPs优化至3.7 G,降幅达43.1%,展现了在内河船舶目标检测精度和效率方面的优势;热力图分析进一步凸显了模型能有效聚焦内河船舶特征,验证了算法在复杂内河场景下的检测鲁棒性.

In inland vessel object detection,many targets fall into the category of small objects,occupying limited pixels in images.Additionally,interference from complex environments often leads to insufficient detection accura-cy,frequent false positives,and missed detections.To address these challenges,this study proposes an object detec-tion algorithm based on PEW-YOLOv8,which integrates YOLOv8 with a P2 detection layer,EfficientNetV2,and the WIoUinner loss function.A new P2 shallow detection layer with a resolution of 160×160 is introduced to en-hance small target detection.A 32-dimensional feature space reconstruction is employed to achieve dynamic weight allocation across multi-scale features.Furthermore,a bidirectional interaction mechanism between high-and low-level features is designed to improve feature extraction for small vessel objects.To address the increased param-eter burden caused by multi-level detection heads,an improved EfficientNetV2 architecture is adopted,which incor-porates a GELU-activated Stem module to mitigate gradient explosion and unstable training.During training,the channel count is expanded fourfold while simplifying the convolutional structure,significantly accelerating the train-ing process without sacrificing model quality.Besides,the WIoUinner loss function with a dynamic non-monotonic focusing mechanism is designed,which introduces auxiliary prediction boxes with varying scales to accelerate the convergence of bounding boxes.When the predicted and ground truth boxes are closed aligned,the model places greater emphasis on the distance between center points,reducing the penalty from geometric metrics and improving generalization capability.The algorithm is validated using a dataset that combines the publicly available Seaships da-taset with a self-constructed inland vessel dataset.Experimental results demonstrate that compared to YOLOv10,PEW-YOLOv8 achieves an average detection accuracy of 94.8%,a 3%improvement.Computational complexity is significantly reduced,with FLOPs optimized to 3.7 G,representing a 43.1%reduction,which demonstrates the mod-el's advantages in both accuracy and efficiency for inland vessel detection tasks.Heatmap analysis further confirms the model's ability to effectively focus on inland vessel features,demonstrating robust detection performance in complex inland waterway scenarios.

曹智远;马勇;成雪夫;胡文韬

武汉理工大学航运学院 武汉 430063||武汉理工大学水路交通控制全国重点实验室 武汉 430063武汉理工大学航运学院 武汉 430063||武汉理工大学水路交通控制全国重点实验室 武汉 430063长江信达软件技术(武汉)有限责任公司 武汉 430079长江信达软件技术(武汉)有限责任公司 武汉 430079

交通工程

智能船舶内河船舶PEW-YOLO目标检测WIoU

intelligent shipsinland vesselsPEW-YOLOobject detectionWIoU

《交通信息与安全》 2025 (2)

36-43,8

国家重点研发计划项目(2023YFB4302300)资助

10.3963/j.jssn.1674-4861.2025.02.005

评论