Laser & Optoelectronics Progress, Volume. 59, Issue 24, 2410005(2022)
Algorithm on Mushroom Image Classification Based on CA-EfficientNetV2
Figures & Tables(15)
Fig. 6. Data enhancement results. (a) Original picture; (b) random resize; (c) padding; (d) random rotation; (e) color jitter; (f) random crop
Fig. 9. Accuracy comparison curve of EfficientNetV2 network before and after improvement
Fig. 10. Loss function comparison curve of NetworkNetV2 network before and after improvement
Fig. 11. Comparison of image visualization results. (a) Input; (b) EfficientNetV2; (c) CA- EfficientNetV2
Table 1. EfficientNetV2-S structure table
View tableTable 1. EfficientNetV2-S structure table
Stage Operator Stride Channels Layers 0 Conv3×3 2 24 1 1 Fused-MBConv1，k3×3 1 24 2 2 Fused-MBConv4，k3×3 2 48 4 3 Fused-MBConv4，k3×3 2 64 4 4 MBConv4，k3×3，SE0.25 2 128 6 5 MBConv6，k3×3，SE0.25 1 160 9 6 MBConv6，k3×3，SE0.25 2 272 15 7 Conv1×1&Pooling&FC 1792 1
Table 2. Experimental parameter settings
View tableTable 2. Experimental parameter settings
Experimental parameter Scene setting Activation function SiLU/Sigmoid Learning initial rate 0.01 Learning decayl rate 0.01 Drop_connect_rate 0.2 Loss function Cross entropy Momentum 0.9 Weight_decay 1×10-4 Optimization SGD Batch size 32 Number of iterations 150 Dropout rate 0.2 Activation function Softmax
Table 3. Structural parameters of improved EfficientNetV2 model
View tableTable 3. Structural parameters of improved EfficientNetV2 model
Stage Kernel size /pooling size Output Input 224×224×3 Conv 3×3 112×112×24 Fused-MBConv 3×3 112×112×24 Fused-MBConv 3×3 56×56×48 Fused-MBConv 3×3 28×28×64 MBConv 3×3 14×14×128 MBConv 3×3 14×14×160 MBConv 3×3 7×7×272 Ave-pooling 3×3 7×7×272 Softmax 1×1 1×1×9
Table 4. Comparison of classification accuracy for different models
View tableTable 4. Comparison of classification accuracy for different models
Model Accuracy ResNet50 86.7 RegNet 91.9 EfficientNetV2 93.8 CA-EfficientNetV2 96.8
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
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)
© Copyright 2018-2021 | Chinese Laser Press.
All Rights Reserved 沪ICP备15018463号-20