SFA-Net: A Selective Features Absorption Network for Object Detection in Rainy Weather Conditions

被引:47
作者
Huang, Shih-Chia [1 ]
Hoang, Quoc-Viet [2 ,3 ]
Le, Trung-Hieu [1 ,3 ]
机构
[1] Natl Taipei Univ Technol, Dept Elect Engn, Taipei 10608, Taiwan
[2] Natl Taipei Univ Technol, Int Grad Program Elect Engn & Comp Sci, Taipei 10608, Taiwan
[3] Hung Yen Univ Technol & Educ, Fac Informat Technol, Hungyen 17000, Vietnam
关键词
Feature extraction; Object detection; Rain; Absorption; Meteorology; Image restoration; Detectors; Convolutional neural network (CNN); object detection; rain removal; SFA-Net;
D O I
10.1109/TNNLS.2021.3125679
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页码:5122 / 5132
页数:11
相关论文
共 46 条
[1]  
[Anonymous], 2018, IMAGE SUPER RESOLUTI
[2]  
Bebber, CODROPS RAINEFFECT R
[3]   Driver-Activity Recognition in the Context of Conditionally Autonomous Driving [J].
Braunagel, Christian ;
Stolzmann, Wolfgang ;
Kasneci, Enkelejda ;
Rosenstiel, Wolfgang .
2015 IEEE 18TH INTERNATIONAL CONFERENCE ON INTELLIGENT TRANSPORTATION SYSTEMS, 2015, :1652-1657
[4]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[5]   The Pascal Visual Object Classes (VOC) Challenge [J].
Everingham, Mark ;
Van Gool, Luc ;
Williams, Christopher K. I. ;
Winn, John ;
Zisserman, Andrew .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2010, 88 (02) :303-338
[6]   Lightweight Pyramid Networks for Image Deraining [J].
Fu, Xueyang ;
Liang, Borong ;
Huang, Yue ;
Ding, Xinghao ;
Paisley, John .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2020, 31 (06) :1794-1807
[7]   Clearing the Skies: A Deep Network Architecture for Single-Image Rain Removal [J].
Fu, Xueyang ;
Huang, Jiabin ;
Ding, Xinghao ;
Liao, Yinghao ;
Paisley, John .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2017, 26 (06) :2944-2956
[8]  
Girshick Ross, 2014, CVPR, DOI DOI 10.1109/CVPR.2014.81
[9]  
Goodfellow IJ, 2014, ADV NEUR IN, V27, P2672
[10]  
HE KM, 2016, PROC CVPR IEEE, P770, DOI [DOI 10.1109/CVPR.2016.90, 10.1109/CVPR.2016.90]