A lightweight convolutional neural network for road surface classification under shadow interference

被引:0
|
作者
Mao, Ruichi [1 ]
Wu, Guangqiang [1 ]
Wu, Jian [1 ]
Wang, Xingyu [1 ]
机构
[1] Tongji Univ, Sch Automot Studies, Shanghai 201804, Peoples R China
基金
中国国家自然科学基金;
关键词
Road classification; Data augmentation; Computer vision; Convolution neural network;
D O I
10.1016/j.knosys.2024.112761
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The development of intelligent driving, especially in the intelligent control of active suspension, heavily relies on the predictive perception of upcoming road conditions. To achieve accurate real-time road surface classification and overcome shadow interference, a lightweight convolutional neural network (CNN) based on a novel data augmentation method is proposed and an improved cycle-consistent adversarial network (CycleGAN) is developed to generate shadowed pavement data. The CycleGAN network structure is optimized using the texture selfsupervised (TSS) mechanism and the learned perceptual image patch similarity (LPIPS) function, with label smoothing applied during training. The images produced by this data augmentation method closely resemble real-world images. Furthermore, Efficient-MBConv, which offers the advantages of fewer parameters and higher precision, is proposed. Finally, the Light-EfficientNet architecture, based on Efficient-MBConv, is developed and trained on the augmented dataset. Compared with EfficientNet-B0, the number of parameters in LightEfficientNet is reduced by 61.94 %. The Light-EfficientNet model trained with data augmentation demonstrates an average classification accuracy improvement of 5.76 % on the test set with shadows, compared with the model trained without data augmentation. This approach effectively reduces the impact of shadows on road classification at a lower cost, while also significantly reducing the computational resources required by the CNN, providing real-time and accurate road surface information for the control of active suspension height and damping.
引用
收藏
页数:12
相关论文
共 50 条
  • [1] A lightweight convolutional neural network for detecting road cracks
    Ren, Xinghua
    Hu, Shaolin
    Hou, Yandong
    Ye, Ke
    Chen, Zhengquan
    Wu, Zhengbo
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (10) : 6729 - 6743
  • [2] A Lightweight Convolutional Neural Network for Hyperspectral Image Classification
    Jia, Sen
    Lin, Zhijie
    Xu, Meng
    Huang, Qiang
    Zhou, Jun
    Jia, Xiuping
    Li, Qingquan
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (05): : 4150 - 4163
  • [3] Application of a Lightweight Convolutional Neural Network in Ship Classification
    Wang Wenliang
    Yang Xiaodi
    Zhang Boya
    Ma Jishun
    Zeng Peng
    Han Peng
    LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (06)
  • [4] Road Surface Classification Based on Radar Imaging Using Convolutional Neural Network
    Sabery, Shahrzad Minooee
    Bystrov, Aleksandr
    Gardner, Peter
    Stroescu, Ana
    Gashinova, Marina
    IEEE SENSORS JOURNAL, 2021, 21 (17) : 18725 - 18732
  • [5] Classification of road surfaces using convolutional neural network
    Balcerek, Julian
    Konieczka, Adam
    Piniarski, Karol
    Pawlowski, Pawel
    2020 SIGNAL PROCESSING - ALGORITHMS, ARCHITECTURES, ARRANGEMENTS, AND APPLICATIONS (SPA), 2020, : 98 - 103
  • [6] A Lightweight Hybrid Convolutional Neural Network for Hyperspectral Image Classification
    Ma, Xiaohu
    Kang, Xudong
    Qin, Huawei
    Wang, Wuli
    Ren, Guangbo
    Wang, Jianbu
    Liu, Baodi
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [7] A Lightweight Conditional Convolutional Neural Network for Hyperspectral Image Classification
    Wu, Linfeng
    Wang, Huajun
    Wang, Huiqing
    PHOTOGRAMMETRIC ENGINEERING AND REMOTE SENSING, 2023, 89 (07): : 413 - 423
  • [8] Lightweight Convolutional Neural Network for Fire Classification in Surveillance System
    Nguyen, Duy-Linh
    Putro, Muhamad Dwisnanto
    Jo, Kang-Hyun
    IEEE ACCESS, 2023, 11 : 101604 - 101615
  • [9] A Lightweight Convolutional Neural Network for Silkworm Cocoons Fast Classification
    Feng, Wei
    Jia, Geng
    Wang, Wei
    Zhang, Zukui
    Cui, Jing
    Chu, Zhongyi
    Xu, Bo
    COGNITIVE SYSTEMS AND SIGNAL PROCESSING, PT II, 2019, 1006 : 301 - 309
  • [10] A Lightweight Convolutional Neural Network for White Blood Cells Classification
    Ridoy, Md Alif Rahman
    Islam, Md Rabiul
    2020 23RD INTERNATIONAL CONFERENCE ON COMPUTER AND INFORMATION TECHNOLOGY (ICCIT 2020), 2020,