交通信息与安全2025，Vol.43Issue(2)：36-43,8.DOI:10.3963/j.jssn.1674-4861.2025.02.005
基于PEW-YOLOv8的内河船舶目标检测方法
A Method for Inland Vessel Object Detection Based on PEW-YOLOv8
摘要
Abstract
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.关键词
智能船舶/内河船舶/PEW-YOLO/目标检测/WIoUKey words
intelligent ships/inland vessels/PEW-YOLO/object detection/WIoU分类
交通工程引用本文复制引用
曹智远,马勇,成雪夫,胡文韬..基于PEW-YOLOv8的内河船舶目标检测方法[J].交通信息与安全,2025,43(2):36-43,8.基金项目
国家重点研发计划项目(2023YFB4302300)资助 （2023YFB4302300）
