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

被引:38
|
作者
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
相关论文
共 50 条
  • [31] Deep Learning-Based Object Detection in Diverse Weather Conditions
    Ravinder, M.
    Jaiswal, Arunima
    Gulati, Shivani
    INTERNATIONAL JOURNAL OF INTELLIGENT INFORMATION TECHNOLOGIES, 2022, 18 (01)
  • [32] Robust 3D Object Detection in Cold Weather Conditions
    Piroli, Aldi
    Dallabetta, Vinzenz
    Walessa, Marc
    Meissner, Daniel
    Kopp, Johannes
    Dietmayer, Klaus
    2022 IEEE INTELLIGENT VEHICLES SYMPOSIUM (IV), 2022, : 287 - 294
  • [33] Features Split and Aggregation Network for Camouflaged Object Detection
    Zhang, Zejin
    Wang, Tao
    Wang, Jian
    Sun, Yao
    JOURNAL OF IMAGING, 2024, 10 (01)
  • [34] Efficient Selective Context Network for Accurate Object Detection
    Nie, Jing
    Pang, Yanwei
    Zhao, Shengjie
    Han, Jungong
    Li, Xuelong
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2021, 31 (09) : 3456 - 3468
  • [35] Selective feature fusion network for salient object detection
    Sun, Fengming
    Yuan, Xia
    Zhao, Chunxia
    IET COMPUTER VISION, 2023, 17 (04) : 483 - 495
  • [36] A fused network based on PReNet and YOLOv4 for traffic object detection in rainy environment
    Chen T.
    Yao D.-C.
    Gao T.
    Qiu H.-H.
    Guo C.-X.
    Liu Z.-W.
    Li Y.-H.
    Bian H.-Y.
    Jiaotong Yunshu Gongcheng Xuebao/Journal of Traffic and Transportation Engineering, 2022, 22 (03): : 225 - 237
  • [37] Yolov4-based hybrid feature enhancement network with robust object detection under adverse weather conditions
    Patil, Shankar M.
    Pawar, Shivaji D.
    Mhatre, Sonali N.
    Kharade, Prakash A.
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (05) : 4243 - 4258
  • [38] QO-Net: Query Optimization Underwater Object Detection Network
    Tian, Jiandong
    Fan, Baojie
    Sun, Hongyang
    Xu, Hongxin
    2024 IEEE/RSJ INTERNATIONAL CONFERENCE ON INTELLIGENT ROBOTS AND SYSTEMS (IROS 2024), 2024, : 8642 - 8649
  • [39] G-Net: An Efficient Convolutional Network for Underwater Object Detection
    Zhao, Xiaoyang
    Wang, Zhuo
    Deng, Zhongchao
    Qin, Hongde
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2024, 12 (01)
  • [40] LKC-Net: large kernel convolution object detection network
    Weina Wang
    Shuangyong Li
    Jiapeng Shao
    Huxidan Jumahong
    Scientific Reports, 13