江苏大学学报(自然科学版)2026,Vol.47Issue(1):48-54,7.DOI:10.3969/j.issn.1671-7775.2026.01.007
基于改进ShuffleNet V2网络的路面类型识别
Road surface type recognition based on improved ShuffleNet V2 network
摘要
Abstract
To solve the problems of pavement type recognition models with large volume and low accuracy,the improved ShuffleNet V2 network pavement type recognition model was proposed.The efficient channel attention(EC A)module was added to the ShuffleNet V2 network structure to achieve cross channel information interaction by attention mechanism.The size of convolution kernel was adjusted according to the number of input channels.The ReLU function was replaced by the LeakyRelu function to avoid the invalidation of activation function.In order to improve the feature extraction ability and generalization ability of the model,the module composed of inflated convolution was introduced to obtain the wider range of image information with the image resolution unchanged.According to the classification characteristics of pavement types,the number of each module stacked and the overall architecture of the network were adjusted to reduce the model's computational and parametric quantities.The improved algorithm was verified on the road surface classification dataset(RSCD).The results show that the enhanced ShuffleNet V2 model achieves parameter quantity of 4.67×106,representing reduction of 1.4 ×105 compared to the original model.The accuracy reaches 95.53%with improvement of 0.71 percentage point over the pre-optimization level.The inference time is reduced by 31%,and the accuracy of road surface type recognition and response speed are improved.关键词
路面类型识别/卷积神经网络/ShuffleNet模型/ECA注意力机制/膨胀卷积模块/轻量化模型Key words
road surface type recognition/convolutional neural network(CNN)/ShuffleNet model/efficient channel attention(ECA)attention mechanism/dilated convolution module/lightweight model分类
信息技术与安全科学引用本文复制引用
张缓缓,冯屹轩,吴宏超..基于改进ShuffleNet V2网络的路面类型识别[J].江苏大学学报(自然科学版),2026,47(1):48-54,7.基金项目
国家自然科学基金资助项目(51705306) (51705306)