Laser & Optoelectronics Progress, Volume. 59, Issue 24, 2410005(2022)
Algorithm on Mushroom Image Classification Based on CA-EfficientNetV2
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.
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
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)