Huang Shih-Chia, Hoang Quoc-Viet, Le Trung-Hieu
IEEE Trans Neural Netw Learn Syst. 2023 Aug;34(8):5122-5132. doi: 10.1109/TNNLS.2021.3125679. Epub 2023 Aug 4.
In recent years, object detection approaches using deep convolutional neural networks (CNNs) have derived major advances in normal images. However, such success is hardly achieved with rainy images due to lack of visibility. Aiming to bridge this gap, in this article, we present a novel selective features absorption network (SFA-Net) to improve the performance of object detection not only in rainy weather conditions but also in favorable weather conditions. SFA-Net accomplishes this objective by utilizing three subnetworks, where the feature selection subnetwork is concatenated with the object detection subnetwork through the feature absorption subnetwork to form a unified model. To promote further advancement in object detection impaired by rain, we propose a large-scale rainy image dataset, named srRain, which contains both synthetic rainy images and real-world rainy images for training and testing purposes. srRain is comprised of 25 900 rainy images depicting diverse driving scenarios in the presence of rain with a total of 181 164 instances interpreting five common object categories. Experimental results display that our SFA-Net reaches the highest mean average precision (mAP) of 77.53% on a normal image set, 62.52% on a synthetic rainy image set, 37.34% on a collected natural rainy image set, and 32.86% on a published real rainy image set, surpassing current state-of-the-art object detectors and the combination of image deraining and object detection models while retaining a high speed.
近年来,使用深度卷积神经网络(CNN)的目标检测方法在正常图像领域取得了重大进展。然而,由于能见度不足,在雨天图像上很难取得这样的成功。为了弥补这一差距,在本文中,我们提出了一种新颖的选择性特征吸收网络(SFA-Net),以不仅在雨天条件下而且在有利天气条件下提高目标检测的性能。SFA-Net通过利用三个子网络来实现这一目标,其中特征选择子网络通过特征吸收子网络与目标检测子网络连接,形成一个统一的模型。为了推动受降雨影响的目标检测的进一步发展,我们提出了一个大规模的雨天图像数据集,名为srRain,它包含合成雨天图像和真实世界的雨天图像,用于训练和测试目的。srRain由25900张雨天图像组成,描绘了有雨情况下的各种驾驶场景,共有181164个实例,诠释了五个常见目标类别。实验结果表明,我们的SFA-Net在正常图像集上达到了最高平均精度(mAP)的77.53%,在合成雨天图像集上为62.52%,在收集的自然雨天图像集上为37.34%,在已发表的真实雨天图像集上为32.86%,超过了当前最先进的目标检测器以及图像去雨和目标检测模型的组合,同时保持了高速。