A lightweight model for identifying types of feed raw material based on improved ShuffleNetV2
Author:
Affiliation:

College of Engineering/Ministry of Agriculture and Rural Affairs Key Laboratory of Smart Farming for Agricultural Animals,Huazhong Agricultural University,Wuhan 430070,China

Clc Number:

S512.2;TP183

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference [21]
  • |
  • Related
  • | | |
  • Comments
    Abstract:

    A lightweight model of ShuffleNetV2-EH with higher accuracy of identification, lower complexity of computation, and suitable for identifying the types of feed raw material based on the lightweight convolutional neural network model ShuffleNetV2 to achieve rapid identification of warehousing feed raw materials and solve the difficulties in manually identifying the types of feed raw materials with similar crushing degree, color, and shape in currently processing and producing the combined feed raw materials. Firstly, the efficient channel attention(ECA) mechanism was introduced into the structure of ShuffleNetV2 network model, which adaptively adjusts channel weights based on input to enhance the ability of network model to percept important features in images of feed raw materials. Secondly, ReLU was replaced with HardSwish activation function to improve the recognition accuracy of the model without adding additional weights and parameters of bias. Finally, the structure of ShuffleNetV2 network model was adjusted to reduce the number of parameters and the complexity of computation in the model on the basis of ensuring the recognition accuracy of model. The results showed that the recognition accuracy of ShuffleNetV2-EH model on image sets from 8 types of feed raw materials was 99.13%, 1.38% higher than that of the original ShuffleNetV2 model. Its accuracy, recall, and F1 score increased by 1.45%, 1.63%, and 1.62 %, respectively. The number of model parameters and floating-point operations decreased by 352 092 and 45.27×106, compared to that of the original model. The overall performance was superior to classical convolutional neural network models including AlexNet, VggNet16, GoogLeNet, and ResNet18. It is indicated that the improved ShuffleNetV2 model well balances the complexity of computation and the recognition accuracy of the model, providing an algorithm foundation for online identification of feed raw materials in the warehousing process.

    Fig.1 Automatic sampling and identification device for feedstuffs
    Fig.2 Schematic diagram of device working status
    Fig.3 Images of some feed material samples
    Fig.4 ShuffleNetV2 unit
    Fig.5 ShuffleNetV2 model structure
    Fig.6 ShuffleNetV2-EH network model structure
    Fig.7 ECA structure diagram
    Fig.8 Loss values of different improved models on the training set(A) and accuracy of different improved models on the validation set(B)
    Fig.9 Confusion matrix of feed ingredient type identification model
    Table 1 The impact of different attention mechanisms on model performance
    Table 2 The impact of different activation functions on model performance
    Table 3 Improved ablation test results of ShuffleNetV2 mode
    Table 4 Comparative test results of different model performances
    Reference
    [1] 牛智有,于重洋,吴志陶,等.基于改进ResNet18模型的饲料原料种类识别方法[J].农业机械学报,2023,54(2):378-385.NIU Z Y,YU C Y,WU Z T,et al.Identification of feed raw material type based on improved ResNet18 model[J].Transactions of the CSAM,2023,54(2):378-385(in Chinese with English abstract).
    [2] 张伟健,牛智有,刘静,等.粉碎机筛网破损在线自动识别装置设计与试验[J].农业工程学报,2021,37(2):61-70.ZHANG W J,NIU Z Y,LIU J,et al. Design and experiments of online automatic identification device for screen breakage of hammer mill[J].Transactions of the CSAE,2021,37(2):61-70(in Chinese with English abstract).
    [3] CHAUDHARY P,CHAUDHARI A K,GODARA S.Color transform based approach for disease spot detection on plant leaf[J].International journal of computer science and telecommunications,2012,3(6):65-70.
    [4] 刘立波,周国民.基于多层感知神经网络的水稻叶瘟病识别方法[J].农业工程学报,2009,25(13):213-217.LIU L B, ZHOU G M.Identification method of rice leaf blast using multilayer perception neural network[J].Transactions of the CSAE,2009,25(13):213-217(in Chinese with English abstract).
    [5] 王献锋,张善文,王震,等.基于叶片图像和环境信息的黄瓜病害识别方法[J].农业工程学报,2014,30(14):148-153.WANG X F, ZHANG S W, WANG Z,et al. Recognition of cucumber diseases based on leaf image and environmental information[J].Transactions of the CSAE,2014,30(14):148-153(in Chinese with English abstract).
    [6] PATLE A,CHOUHAN D S.SVM kernel functions for classification[C]//2013 International Conference on Advances in Technology and Engineering (ICATE),January 23-25,2013,Mumbai.Mumbai:IEEE,2013:1-9.
    [7] ZHANG S C,LI X L,ZONG M,et al.Efficient kNN classification with different numbers of nearest neighbors[J].IEEE transactions on neural networks and learning systems,2018,29(5):1774-1785.
    [8] GOVARDHAN M,VEENA M B.Diagnosis of tomato plant diseases using random forest[C]//2019 Global Conference for Advancement in Technology (GCAT),October 18-20,2019,Bangaluru,India.Bangaluru:IEEE,2019:1-5.
    [9] KHALID S,KHALIL T,NASREEN S.A survey of feature selection and feature extraction techniques in machine learning[C]//2014 Science and Information Conference,August 27-29,2014,London,UK.London:IEEE,2014:372-378.
    [10] SZEGEDY C,LIU W,JIA Y Q,et al.Going deeper with convolutions[C]//2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR),June 7-12,2015,Boston,MA,USA.Boston:IEEE,2015:1-9.
    [11] HE K M,ZHANG X Y,REN S Q,et al.Deep residual learning for image recognition[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR),June 27-30,2016,Las Vegas,NV,USA.Las Vegas:IEEE,2016:770-778.
    [12] HUANG G,LIU Z,VAN DER MAATEN L,et al.Densely connected convolutional networks[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR),July 21-26,2017,Honolulu,HI.Honolulu:IEEE,2017:4700-4708.
    [13] 李恩霖,谢秋菊,苏中滨,等.基于深度学习的玉米叶片病斑识别方法研究[J].智慧农业导刊,2021,1(10):1-10.LI E L,XIE Q J,SU Z B,et al.Research on corn disease identification method based on deep learning[J].Journal of smart agriculture,2021,1(10):1-10(in Chinese with English abstract).
    [14] LUO Y Q,SUN J,SHEN J F,et al.Apple leaf disease recognition and sub-class categorization based on improved multi-scale feature fusion network[J].IEEE access,2021,9:95517-95527.
    [15] 孙俊, 朱伟栋, 罗元秋,等.基于改进MobileNet-V2的田间农作物叶片病害识别[J].农业工程学报, 2021, 37(22): 161-169. SUN J, ZHU W D, LUO Y Q,et al. Recognizing the diseases of crop leaves in fields using improved Mobilenet-V2[J].Transactions of the CSAE, 2021,37(22):161-169(in Chinese with English abstract).
    [16] 彭红星,何慧君,高宗梅,等.基于改进ShuffleNetV2模型的荔枝病虫害识别方法[J].农业机械学报,2022,53(12):290-300. PENG H X,HE H J,GAO Z M,et al. Litchi diseases and insect pests identification method based on improved ShuffleNetV2[J].Transactions of the CSAM,2022,53(12):290-300(in Chinese with English abstract).
    [17] 张鹏程,余勇华,陈传武,等.基于改进MobileNetV2的柑橘害虫分类识别方法[J].华中农业大学学报,2023,42(3):161-168. ZHANG P C,YU Y H, CHEN C W,et al.A classification and recognition method for citrus insect pests based on improved MobileNetV2[J].Journal of Huazhong Agricultural University,2023,42(3):161-168(in Chinese with English abstract).
    [18] MA N N,ZHANG X Y,ZHENG H T,et al.ShuffleNet V2:practical guidelines for efficient CNN architecture design[M]//Anon.Lecture notes in computer science.Cham:Springer International Publishing,2018:122-138.
    [19] WANG Q L,WU B G,ZHU P F,et al.ECA-net:efficient channel attention for deep convolutional neural networks[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR),June 13-19,2020,Seattle,WA,USA.Seattle:IEEE,2020:11534-11542.
    [20] HOWARD A,SANDLER M,CHEN B,et al.Searching for MobileNetV3[C]//2019 IEEE/CVF International Conference on Computer Vision (ICCV),October 27-November 2,2019,Seoul,Korea.Seoul:IEEE,2019:1314-1324.
    [21] BANAN A,NASIRI A,TAHERI-GARAVAND A.Deep learning-based appearance features extraction for automated carp species identification[J/OL].Aquacultural engineering,2020,89:102053[2024-07-11].https://doi.org/10.1016/j.aquaeng.2020.102053.
    Related
    Cited by
Get Citation

田敏,牛智有,刘梅英. A lightweight model for identifying types of feed raw material based on improved ShuffleNetV2[J]. Jorunal of Huazhong Agricultural University,2025,44(2):105-115.

Copy
Share
Article Metrics
  • Abstract:14
  • PDF: 35
  • HTML: 0
  • Cited by: 0
History
  • Received:July 11,2024
  • Online: April 02,2025
Article QR Code