Succulent Plant Image Classification Based on Lightweight GoogLeNet with CBAM Attention Mechanism
Aiming at the model overfitting problem caused by limited datasets and visual complexity in succulent plant classification tasks, this study proposes a GoogLeNet classification method based on lightweighting and improving the Convolutional Block attention module (CBAM). Meanwhile, batch normalizatio...
Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2025-03-01
|
| Series: | Applied Sciences |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2076-3417/15/7/3730 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| Summary: | Aiming at the model overfitting problem caused by limited datasets and visual complexity in succulent plant classification tasks, this study proposes a GoogLeNet classification method based on lightweighting and improving the Convolutional Block attention module (CBAM). Meanwhile, batch normalization (BN) operations are added after each convolutional layer to accelerate network convergence and improve model stability. In addition, the model’s ability to extract key features is enhanced by integrating the channel and spatial attention mechanisms of the CBAM attention module. Experimental results show that the improved lightweight GoogLeNet achieves 99.4% classification accuracy on the validation set, effectively mitigates the overfitting problem, and maintains high computational efficiency in resource-constrained environments. The model parameters and computational complexity are significantly reduced by streamlining the Inception modules from nine to seven and introducing depth-separable convolution. To further validate the model robustness, this study extends the dataset via data augmentation methods, and the experiments show that the improved model still maintains stable performance in small dataset environments, demonstrating its advantages in data-scarce scenarios. This study provides an effective solution for the task of succulent plant classification, which has significant application value. Future research will focus on further optimization of the model structure to continuously improve the classification accuracy and robustness. |
|---|---|
| ISSN: | 2076-3417 |