College of Information Science and Technology, Donghua University, Shanghai 201620, China.
Sensors (Basel). 2022 Oct 13;22(20):7786. doi: 10.3390/s22207786.
An improved maritime object detection algorithm, SRC-YOLO, based on the YOLOv4-tiny, is proposed in the foggy environment to address the issues of false detection, missed detection, and low detection accuracy in complicated situations. To confirm the model's validity, an ocean dataset containing various concentrations of haze, target angles, and sizes was produced for the research. Firstly, the Single Scale Retinex (SSR) algorithm was applied to preprocess the dataset to reduce the interference of the complex scenes on the ocean. Secondly, in order to increase the model's receptive field, we employed a modified Receptive Field Block (RFB) module in place of the standard convolution in the Neck part of the model. Finally, the Convolutional Block Attention Module (CBAM), which integrates channel and spatial information, was introduced to raise detection performance by expanding the network model's attention to the context information in the feature map and the object location points. The experimental results demonstrate that the improved SRC-YOLO model effectively detects marine targets in foggy scenes by increasing the mean Average Precision (mAP) of detection results from 79.56% to 86.15%.
提出了一种改进的海上目标检测算法 SRC-YOLO,它基于 YOLOv4-tiny,旨在解决复杂情况下的误检、漏检和低检测精度问题。为了验证模型的有效性,研究中生成了一个包含各种浓度雾、目标角度和大小的海洋数据集。首先,应用单尺度反射率(SSR)算法对数据集进行预处理,以减少复杂场景对海洋的干扰。其次,为了增加模型的感受野,我们在模型的 Neck 部分使用了改进的 Receptive Field Block(RFB)模块来替代标准卷积。最后,引入卷积块注意力模块(CBAM),通过将网络模型的注意力扩展到特征图中的上下文信息和目标位置点,提高检测性能。实验结果表明,改进后的 SRC-YOLO 模型通过将检测结果的平均精度(mAP)从 79.56%提高到 86.15%,有效地检测了雾天场景中的海洋目标。