• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

计算机工程与科学

• • 上一篇    下一篇

夜视海面红外舰船目标轻量化检测算法

黄建硕, 赵宇清, 王 佳, 徐保勇   

  1. (1.湖南科技大学物理与电子科学学院,湖南 湘潭  411201;
    2.长沙学院电子信息与电气工程学院,湖南 长沙  410022;
    3.武汉中和海洋光讯有限公司,湖北 武汉  430070)

Night Vision Infrared Surface Ship Target Lightweight Detection Algorithm

HUANG Jianshuo, ZHAO Yuqing, WANG Jia, Xu Baoyong   

  1. (1. School of Physics and Electronic Science, Hunan University of Science and Technology, Xiangtan  411201, China;
    2. School of Electronic Information and Electrical Engineering, Changsha University, Changsha  410022, China;
    3. Wuhan Zhonghe Ocean Optical Communication Co., Ltd., Wuhan  430070, China) 

摘要: 针对夜间复杂海况下红外舰船检测任务中存在的图像对比度低、目标特征弱、远距离小目标易漏检以及现有检测模型参数冗余等问题,提出一种多尺度轻量化目标检测网络——GRBP-YOLO。首先,将组混洗卷积(GSConv)模块替换传统的卷积模块,利用通道分组与特征重排机制实现计算效率与表征能力的平衡,从而在维持检测精度的前提下显著降低模型复杂度;其次,提出残差分解跨阶段部分融合模块(RS_C2f),通过残差连接与分解堆叠结构的协同设计,在继承C2f模块轻量化优势的基础上,强化跨层级特征交互能力,并进一步减少参数冗余;最后,构建双向特征金字塔网络(BiFPN)并融合P2层小目标检测头,增强通过跨尺度特征重校准机制提升微小目标检测性能,有效地解决远距离微小舰船目标因对比度较低、分辨率不足等导致的漏检问题。实验结果表明,在夜间红外船舶等数据集中相较于基线模型YOLOv8n,GRBP-YOLO算法平均检测精度(mAP@0.5)和召回率分别提升了1.5%、1.9%,达到了92.5%、88%,模型参数量降至1.7M,文件大小仅3.9MB。实验验证了GRBP-YOLO算法的有效性,为复杂海况下的实时舰船检测提供了新的技术解决方案。

关键词: 红外舰船检测, YOLOv8, GSConv, BiFPN, 图像处理

Abstract: To address the challenges of low image contrast, weak target features, missed detections of small distant targets, and the redundancy of existing detection models in infrared ship detection under complex nighttime sea conditions, this paper proposes a multi-scale lightweight object detection network—GRBP-YOLO. First, the Group Shuffle Convolution (GSConv) module is introduced to replace the traditional convolutional layers. By leveraging channel grouping and feature rearrangement mechanisms, GSConv achieves a balance between computational efficiency and representational capability, significantly reducing model complexity while maintaining detection accuracy. Second, a Residual Split Cross-Stage Partial Fusion Module (RS_C2f) is proposed. Through the synergistic design of residual connections and a split-stacking structure, RS_C2f strengthens cross-layer feature interaction while inheriting the lightweight advantages of the original C2f module, further reducing parameter redundancy. Finally, a Bidirectional Feature Pyramid Network (BiFPN) is constructed, incorporating an additional P2-level small object detection head. This design enhances the detection of tiny targets by recalibrating multi-scale features, effectively mitigating the missed detections of small, low-contrast ship targets at long distances caused by limited resolution. Experimental results show that, compared to the baseline model YOLOv8n, the GRBP-YOLO algorithm achieves an average improvement of 1.5% and 1.9% in detection accuracy (mAP@0.5) and recall rate, respectively, reaching 92.5% and 88%. The model parameter count is reduced to 1.7 million, and the file size is only 3.9 MB. The experiments validated the effectiveness of the GRBP-YOLO algorithm, providing a new technical solution for real-time ship detection in complex maritime conditions.

Key words: Infrared ship detection, YOLOv8, GSConv, BiFPN, Image proceessing