An improved version of the EfficientNetV2 network is presented for garbage image classification to address the limitations of mainstream algorithms, such as poor dataset universality, limited recognition types, and algorithmic constraints in specific environments. The proposed algorithm emphasized both classification speed and accuracy. The EfficientNetV2 network was utilized as the baseline model, and classification speed was enhanced through the incorporation of the SK (selective kernel) attention mechanism. Transfer learning strategies were employed to improve classification accuracy. By leveraging deep learning model frameworks for garbage image processing, the need for manual feature extraction from dataset images was eliminated, and the scope of garbage recognition was expanded. Experimental results demonstrate that the proposed algorithm achieves an accuracy of 99.71% on a self-built dataset, which is an improvement of at least 4.77% compared to other algorithms, such as GoogleNet. Furthermore, in terms of time efficiency, the proposed algorithm outperforms algorithms like VggNet19 by at least 50%. Through the enhancement of the EfficientNetV2 network, accurate and faster garbage classification is enabled, providing a scientific and efficient solution to the growing challenges posed by garbage issues.
| 科 Family | 属数 Number of genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) | 属 Genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) |
|---|---|---|---|---|---|---|
| 鹅膏菌科Amanitaceae | 2 | 11 | 5.26 | 鹅膏菌属 Amanita | 10 | 4.78 |
| 小菇科 Mycenaceae | 2 | 12 | 5.74 | 丝盖伞属 Inocybe | 5 | 2.39 |
| 多孔菌科 Polyporaceae | 8 | 14 | 6.70 | 蜡蘑属 Laccaria | 5 | 2.39 |
| 红菇科 Russulaceae | 3 | 23 | 11.00 | 小皮伞属 Marasmius | 6 | 2.87 |
| 小菇属 Mycena | 11 | 5.26 | ||||
| 光柄菇属 Pluteus | 5 | 2.39 | ||||
| 红菇属 Russula | 17 | 8.13 | ||||
| 栓菌属 Trametes | 5 | 2.39 |