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 条
  • [21] An improved YOLOv8-based vehicle detection model for rainy weather conditions
    Lin, Dapeng
    Qu, Licheng
    Yue, Wenqi
    Wang, Jian
    Yu, Yanjie
    2024 9TH INTERNATIONAL CONFERENCE ON COMPUTER AND COMMUNICATION SYSTEMS, ICCCS 2024, 2024, : 50 - 55
  • [22] Thermal Object Detection in Difficult Weather Conditions Using YOLO
    Kristo, Mate
    Ivasic-Kos, Marina
    Pobar, Miran
    IEEE ACCESS, 2020, 8 : 125459 - 125476
  • [23] Moving Object Detection Under Rain and Snow Weather Conditions
    Yang Guoliang
    Yu Dingling
    Wang Yang
    Wang Yanfang
    LASER & OPTOELECTRONICS PROGRESS, 2020, 57 (24)
  • [24] ARODNet: adaptive rain image enhancement object detection network for autonomous driving in adverse weather conditions
    Qiu, Yongsheng
    Lu, Yuanyao
    Wang, Yuantao
    Jiang, Haiyang
    OPTICAL ENGINEERING, 2023, 62 (11) : 118101
  • [25] Object detection by selective integration of HLAC mask features
    Hidaka, Akinori
    Kurita, Takio
    Otsu, Nobuyuki
    BLISS 2008: 2008 ECSIS SYMPOSIUM ON BIO-INSPIRED, LEARNING AND INTELLIGENT SYSTEMS FOR SECURITY, PROCEEDINGS, 2008, : 46 - +
  • [26] Object detection network based on dense dilated encoder net
    Liu, Shaohua
    Yang, Ao
    She, Chundong
    Du, Kang
    IET IMAGE PROCESSING, 2023, 17 (09) : 2640 - 2648
  • [27] Bi-Branch Multiscale Feature Joint Network for ORSI Salient Object Detection in Adverse Weather Conditions
    Yuan, Jianjun
    Zou, Xu
    Xia, Haobo
    Liu, Tong
    Wu, Fujun
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [28] DSNet: Joint Semantic Learning for Object Detection in Inclement Weather Conditions
    Huang, Shih-Chia
    Le, Trung-Hieu
    Jaw, Da-Wei
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2021, 43 (08) : 2623 - 2633
  • [29] Who Cares about the Weather? Inferring Weather Conditions for Weather-Aware Object Detection in Thermal Images
    Johansen, Anders Skaarup
    Nasrollahi, Kamal
    Escalera, Sergio
    Moeslund, Thomas B.
    APPLIED SCIENCES-BASEL, 2023, 13 (18):
  • [30] Image-Adaptive YOLO for Object Detection in Adverse Weather Conditions
    Liu, Wenyu
    Ren, Gaofeng
    Yu, Runsheng
    Guo, Shi
    Zhu, Jianke
    Zhang, Lei
    THIRTY-SIXTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FOURTH CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE / THE TWELVETH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2022, : 1792 - 1800