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 条
  • [21] CfRNet: A Lightweight Convolutional Neural Network Classification Model for Rock Image
    Tao, Liuyi
    Li, Xiaochuan
    Li, Jiaqi
    Wang, Jinyi
    2024 5TH INTERNATIONAL CONFERENCE ON COMPUTER ENGINEERING AND APPLICATION, ICCEA 2024, 2024, : 1013 - 1018
  • [22] Nondestructive Classification of Defects in Potatoes Based on Lightweight Convolutional Neural Network
    Yang S.
    Feng Q.
    Zhang J.
    Wang G.
    Zhang P.
    Yan H.
    Shipin Kexue/Food Science, 2021, 42 (10): : 284 - 289
  • [23] Lightweight Spatial Pyramid Convolutional Neural Network for Traffic Sign Classification
    Rachmadi, Reza Fuad
    Koutaki, Gou
    Ogata, Kohichi
    2018 INDONESIAN ASSOCIATION FOR PATTERN RECOGNITION INTERNATIONAL CONFERENCE (INAPR), 2018, : 23 - 28
  • [24] Lightweight Convolutional Neural Network Based on Singularity ROI for Fingerprint Classification
    Jian, Wen
    Zhou, Yujie
    Liu, Hongming
    IEEE ACCESS, 2020, 8 : 54554 - 54563
  • [25] Lightweight convolutional neural network with knowledge distillation for cervical cells classification
    Chen, Wen
    Gao, Liang
    Li, Xinyu
    Shen, Weiming
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 71
  • [26] The classification and localization of crack using lightweight convolutional neural network with CBAM
    Chen, Liujie
    Yao, Haodong
    Fu, Jiyang
    Ng, Ching Tai
    ENGINEERING STRUCTURES, 2023, 275
  • [28] Automatic Recognition of Road Damage Based on Lightweight Attentional Convolutional Neural Network
    Liang, Han
    Lee, Seong-Cheol
    Seo, Suyoung
    SENSORS, 2022, 22 (24)
  • [29] Audio Scene Classification Under Convolutional Neural Network
    Chen, Shengbo
    Tan, Aoshuai
    Liu, Ximin
    Tang, Pengjie
    Lv, Jingxiang
    Guo, Chen
    PROCEEDINGS OF INTERNATIONAL CONFERENCE ON ALGORITHMS, SOFTWARE ENGINEERING, AND NETWORK SECURITY, ASENS 2024, 2024, : 329 - 334
  • [30] Road Crack Classification Based on Improved VGG Convolutional Neural Network
    Cui, Hua
    Liu, Xingwang
    Han, Lixin
    Wei, Zefa
    FUZZY SYSTEMS AND DATA MINING V (FSDM 2019), 2019, 320 : 542 - 547