基于卷积神经网络的生活垃圾图像分类模型设计

摘要

利用基于深度学习的VGG-16卷积神经网络模型,对涵盖4个大类243个小类生活垃圾的77656张图像进行分类检测识别.为了保证模型检测的准确性,先引入Retinex算法对图像进行增强处理,然后对分类交叉熵损失函数进行改进,加入L1正则化和L2正则化,构建新的损失函数.实验结果显示,使用原VGG-16网络模型对垃圾图像进行分类,准确率为94.43％,损失值为0.53;引入Retinex算法对图像进行增强处理后,准确率为95.56％,损失值为0.36;在损失函数中加入L1或者L2正则化后,分类准确率较原网络模型有较小幅度的提升;采用改进后的交叉熵损失函数之后,准确率达97.67％,损失值仅为0.18.