基于改进EfficientNet V1的可回收垃圾分类算法  

作　　者：李镇宏 刘心中 林泽宇 聂文峰 马树森 葛焰 赵津 

机构地区：[1]福建工程学院生态环境与城市建设学院,福建福州350118

出　　处：《物联网技术》2024年第6期79-83,共5页Internet of things technologies

基　　金：国家重点研发计划(2019YFC1904103)。

摘　　要：随着城市的快速发展,居民的生活质量日渐提高,但生活垃圾数量也迅速增加,许多城市都面临着垃圾分类处理的难题。可回收垃圾占生活垃圾的比例很大,且其回收价值较高,可二次利用。针对目前人工分拣和机械分拣低效率、高成本的问题,文中提出了改进EfficientNet V1的可回收垃圾图像分类方法,使用通道和空间注意力模块和空间转换器模块用于改进模型,增强对特征的提取能力。实验结果表明,仅使用空间转换器模块的模型在TrashNet数据集上的准确率达到了98.88%,高于所有对比模型,而使用通道和空间注意力模块以及同时使用两种模块的模型相对于原模型来说精度有提升。在2种公共数据集Cifar100和ImageNet100上进行实验,仅使用空间转换器模块的模型仍获得了最高准确率62.78%和68.17%。利用深度学习技术,实现了可回收垃圾准确快速分类,为垃圾分类事业提供了帮助。

关 键 词：卷积神经网络 垃圾分类 深度学习 注意力机制 图像分类 数据增强 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]