基于改进的ShuffleNet网络面向图片的识别方法

IMAGE ORIENTED RECOGNITION METHOD BASED ON IMPROVED SHUFFLENET NETWORK

  • 摘要: 传统的图像识别往往通过结构复杂且算力高的大型卷积网络实现,在移动式嵌入设备越来越普及的当下,轻量化卷积网络具有参数量少、计算复杂度小、模型能够部署应用于小型设备等特点。基于ShuffleNetV2单元改变网络宽度的同时引入SENet注意力机制组成新的网络模块搭建改进的Shuffle-K5-SENet网络。该网络在保留轻量化的同时,能够更高效地提取图片的特征信息。实验结果表明,在FlowersRecognition数据集上与ShuffleNetV2网络对比,准确度提升了1.88百分点,Loss值下降了3%;在CIFAR-100数据集上与MobileNetV2网络对比,准确度提高了1.84百分点,FLOPs降低23.89%,参数量下降55.82%。

     

    Abstract: Traditional image recognition is often realized by large convolutional networks with complex structure and high computational power. With the increasing popularity of mobile embedded devices, lightweight convolutional networks have the characteristics of less parameters, less computational complexity, and the model can be deployed and applied to small devices. Based on the ShuffleNetV2 unit, the network width was changed and SENet attention mechanism was introduced to form a new network module to build an improved Shufflet-K5-SENet network. The network could extract feature information more efficiently while retaining the lightweight. The experimental results show that compared with ShuffleNetV2 network on Flowers Recognition dataset, the accuracy is improved by 1.88 percentage points and the Loss value is decreased by 3%. Compared with MobileNetV2 network on CIFAR-100 dataset, the accuracy is improved by 1.84 percentage points, the FLOPs is reduced by 23.89% and the number of parameters is reduced by 55.82%.

     

/

返回文章
返回