首页|基于迁移学习的多场景垃圾图像分类方法

基于迁移学习的多场景垃圾图像分类方法

扫码查看
原文链接
  • 万方数据
  • 维普
垃圾分类回收是资源循环再生链条中重要的一环,传统的人工分类方法效率低,而基于神经网络模型的图像识别技术是实现智能化分类回收的重要策略,具有准确度高、泛化性能好等优点.针对生活垃圾场景的多样性以及垃圾图像数据集缺乏的问题,该文在基于单一背景和复杂背景两种不同垃圾图像场景下,提出一种基于预训练ResNeXt101模型进行迁移学习的方法.利用大规模ImageNet图像数据集训练模型参数,并将参数迁移到垃圾图像数据分类模型中进行训练,解决缺乏大规模标注数据的垃圾图像数据集的分类问题,实现玻璃和纸类等4种可回收物的分类识别以及厨余垃圾中杂质的检测.实验结果表明,利用该模型在真实背景下的垃圾图像分类准确率可以达到87.42％,单一背景下分类准确率达到96.19％,实现了高精度的可回收物识别分类以及厨余垃圾中高效的杂质检测,同时有效提升了模型训练速度.
Multi-scene Garbage Image Classification Method Based on Transfer Learning

彭治、刘杨、杜永萍、常燕青、韩红桂

展开 >

北京工业大学 信息学部,北京 100124

江苏省固体废弃物处理环保装备工程技术研究中心,江苏 常州 213000

图像识别 迁移学习 残差网络 RexNeXt 垃圾分类

国家重点研发计划国家重点研发计划

2019YFC19060022018YFC1900804

2022

10.3969/j.issn.1673-629X.2022.05.018

计算机技术与发展
陕西省计算机学会

计算机技术与发展

CSTPCD
影响因子：0.621
ISSN：1673-629X
年,卷(期)：2022.32(5)
  • 1
  • 5