首页|期刊导航|江苏大学学报(自然科学版)|基于改进ShuffleNet V2网络的路面类型识别

基于改进ShuffleNet V2网络的路面类型识别OA

Road surface type recognition based on improved ShuffleNet V2 network

中文摘要英文摘要

针对路面类型识别模型体积大、精确度低的问题,提出基于改进ShuffleNet V2网络的路面类型识别模型.在ShuffleNet V2网络结构中添加高效通道注意力(ECA)模块,通过注意力机制实现跨通道信息交互,并能根据输入的通道数量调整卷积核的大小;使用LeakyRelu函数替代ReLU函数,避免激活函数失效;引入由膨胀卷积组成的模块,在图像分辨率不变的同时,获取更广泛的图像信息,以提高模型的特征提取能力及泛化能力;根据路面类型的分类特点,调整各个模块的堆叠次数和网络的整体架构,降低模型的运算量和参数量.将改进后的算法在道路表面分类数据集(RSCD)上进行验证.结果表明:改进后的ShuffleNet V2模型参数量为4.67×106个,比原模型减少了1.4× 105个;准确率为95.53%,比改进前提高了0.71百分点;推理时间减少了31%,新模型提高了对路面类型识别的准确率和响应速度.

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.

张缓缓;冯屹轩;吴宏超

上海工程技术大学机械与汽车工程学院,上海 201620上海工程技术大学机械与汽车工程学院,上海 201620上海工程技术大学机械与汽车工程学院,上海 201620

信息技术与安全科学

路面类型识别卷积神经网络ShuffleNet模型ECA注意力机制膨胀卷积模块轻量化模型

road surface type recognitionconvolutional neural network(CNN)ShuffleNet modelefficient channel attention(ECA)attention mechanismdilated convolution modulelightweight model

《江苏大学学报(自然科学版)》 2026 (1)

48-54,7

国家自然科学基金资助项目(51705306)

10.3969/j.issn.1671-7775.2026.01.007

评论