激光与红外

作者：张俊

单位：中国电子科技集团公司第五十八研究所,江苏 无锡 214035；

关键词：红外船舶检测;YOLOv8s;模型剪枝;注意力机制

针对海上红外船舶目标检测存在特征不明显、背景干扰复杂、信噪比低等难点,本文提出一种基于改进YOLOv8s的海上红外船舶检测算法。通过在主干网络中引入 BoT模块替换了部分 C2f 结构以更好地捕捉图像中的全局和局部更丰富的信息,提高模型的特征提取能力,同时降低计算开销。并将改进的通道注意力机制模块加入到模型中,用于解决浅层特征图中的语义信息混乱和感受野过小的问题,以提高小目标检测能力。为了减少前向推理的时延引入剪枝技术实现模型剪枝,在保证召回率不降低的同时减少模型的参数。实验表明改进的YOLOv8s海上目标检测算法在相同红外船舶测试数据集上比原始YOLOv8s的AP提高了135,达到了9141。本文所提算法平衡了模型精度与耗时的矛盾关系,在提升性能的同时,保证了模型检测的速度。

来源：2025年第8期

《激光与红外》期刊编辑部

查看激光与红外杂志2025年第8期