高级检索

    基于YOLOv8s剪枝与动态加权ByteTrack的舰载光电船舶跟踪算法

    Shipborne Electro-Optical Ship Tracking Algorithm Based on YOLOv8s Pruning and Dynamically Weighted ByteTrack

    • 摘要: 针对传统舰载光电跟踪系统模型冗余难部署、遮挡场景跟踪连续性差的核心问题,提出基于YOLOv8s剪枝与改进ByteTrack的舰载船舶跟踪算法:先对YOLOv8s做场景适配剪枝以精简模型;再改进ByteTrack,引入双时序关联机制并融合卡尔曼滤波提升遮挡场景预测精度,最后优化加权融合策略强化目标匹配稳定性。基于SeaShips舰载光电场景数据集验证,改进算法mAP50达92.9%、MOTA 82.3%、遮挡恢复成功率91.2%;较原方案,总参数量从11.2 M压缩至7.8 M、MOTA提升4.2%,满足舰载光电系统“实时监测+快速部署”需求。

       

      Abstract: Addressing the issues of model redundancy leading to deployment difficuties and poor tracking continuity in occlusion scenarios in traditional shipborne electro-optical tracking systems, a ship tracking algorithm based on YOLOv8s pruning and improved ByteTrack is proposed in this paper. Firstly, pruning processing is performed on YOLOv8s to simplify redundant structures and reduce the model size. Then, the ByteTrack tracker is improved. A dual-temporal association mechanism combining inter-frame feature persistence and historical trajectory backtracking is introduced, and Kalman filter is integrated to improve the state prediction accuracy in occluded scenarios. Finally, the fusion strategy of temporal features and geometric matching is optimized through weight coefficient adjustment to enhance the stability of target matching. Verified based on the SeaShips dataset covering typical shipborne electro-optical scenarios, the results show that the improved algorithm achieves a mAP50 of 92.9%, MOTA of 82.3%, and an occlusion recovery success rate of 91.2%. Compared with the YOLOv8s+ByteTrack scheme, the effective parameter count (non-zero weights) is reduced by 30%, the total parameter count is reduced by about 30% (from 11.2 M to 7.8 M), the model file size is reduced by 11.8%, and MOTA is increased by 4.2%, which fully meets the core requirements of "real-time monitoring + rapid deployment" for shipborne electro-optical systems.

       

    /

    返回文章
    返回