Laser & Optoelectronics Progress, Volume. 59, Issue 24, 2410005(2022)

Algorithm on Mushroom Image Classification Based on CA-EfficientNetV2

Lisha Meng, Xianzhao Yang*, and Huikang Liu
Author Affiliations
  • Engineering Research Center for Metallurgical Automation and Measurement Technology of Ministry of Education, School of Information Science and Engineering, Wuhan University of Science and Technology, Wuhan 430081, Hubei, China
  • show less

    In view of the low efficiency and poor effect of the traditional mushroom feature extraction method, a lightweight mushroom image classification model is proposed. In view of the small dataset used in the experiment, this classification model initializes the EfficientNetV2 model and modifies the full connection layer in the migration study based on the Imagenet dataset. At the same time, in order to reduce the parameter influence in the network, the original EfficientNetV2 model is streamlined to remove duplicate modules in the network. Finally, the squeeze-and-excitation mechanism in the original MBConv module is replaced with the coordinate attention (CA) attention mechanism with better feature extraction effect, and the new CA-EfficientNetV2 network is obtained. The experimental results show that compared with the classical ResNet50 model and RegNet, the classification accuracy of the proposed EfficientNetV2 is improved by about 10 percentage points and 2 percentage points respectively, and higher generalization performance is obtained; compared with the original EfficientNetV2, the classification accuracy is improved by 3 percentage points. That is, CA-EfficientNetV2 has higher accuracy and classification performance in mushroom classification.

    Tools

    Get Citation

    Copy Citation Text

    Lisha Meng, Xianzhao Yang, Huikang Liu. Algorithm on Mushroom Image Classification Based on CA-EfficientNetV2[J]. Laser & Optoelectronics Progress, 2022, 59(24): 2410005

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Sep. 17, 2021

    Accepted: Oct. 25, 2021

    Published Online: Jan. 11, 2023

    The Author Email: Yang Xianzhao (yangxianzhao@wust.edu.cn)

    DOI:10.3788/LOP202259.2410005

    Topics