基于卷积神经网络的垃圾自动分类算法

全球城市生活垃圾的产生量急剧增长,垃圾分类工作成为十分重要的任务。设计了基于卷积神经网络的2个轻量级模型MobileNetV2与ShuffleNetV1的垃圾分类算法,对MobileNetV2算法进行优化改进,通过调整学习率、选择合适的模型优化器以提高MobileNetV2模型性能。实验结果表明,改进后的模型性能有所提升。