• CN: 11-2187/TH
  • ISSN: 0577-6686

Journal of Mechanical Engineering ›› 2022, Vol. 58 ›› Issue (24): 211-222.doi: 10.3901/JME.2022.24.211

Previous Articles     Next Articles

Dual Attention Network for the Classification of Road Surface Conditions Based on EfficientNet

YANG Wei1, HUANG Lihong1, QU Xiaolei2   

  1. 1. School of Automobile, Chang'an University, Xi'an 710064;
    2. School of Instrumentation and Optoelectronics Engineering, Beihang University, Beijing 100191
  • Received:2022-02-01 Revised:2022-07-05 Online:2022-12-20 Published:2023-04-03

Abstract: Given the drawback of information loss of high-level features caused by convolution operations when the existing EfficientNet is applied to classify asphaltroad surfaces, a novel dual attention mechanism combined two types of attention modules, named channel attention module and position attention module respectively, is introduced to the existing EfficientNet, and the dual attention network based on EfficientNet (DAEfficientNet) is proposed using sigmoid linear unit (SiLU) activation functions and a cosine learning rate decay technique. First, a dataset including 5,938 images of five types of asphalt road surfaces under various weather conditions is constructed. The snow image samples of asphalt road surfaces are from the open-source dataset named Canadian adverse driving conditions dataset (CADCD). Second, the proposed model is trained and the image classification results are produced. Finally, the accuracy, precision, recall, F1 score, and specificity of the analyzed models are calculated to compare the classification performance between the proposed model and the others previous convolutional neural network models. The experimental results show that the proposed method outperforms the others completing methods and achieves higher accuracy and stronger robustness in the task of classification of the five types of road surface images.

Key words: convolutional neural network, dual attention mechanism, asphalt road, road classification

CLC Number: