摘 要
垃圾分类识别系统,通过机器学习库tensorflow作为模型构建框架,使用CNN卷积神经网络构建模型,并通过对数据集的处理划分测试集和训练集,通过多轮迭代得到训练好的模型,再将模型进行封装,并开发一个WEB界面系统用于用户的操作,最后实现用户在WEB页面中输入一张垃圾分类图片,系统以弹窗的形式显示该垃圾的信息
项目运行结果
1、系统首页
2、识别界面
3、垃圾分类识别结果
4、数据管理界面
垃圾分类识别系统主要界面分为三个,系统说明首页、用户输入图片检测页面、管理人员后台信息管理页面
通过启动Django应用程序,打开网页输入127.0.0.1:8000可进入系统说明首页。
二、算法模块实现
本模块主要实现对垃圾分类识别系统中模型训练部分。通过对数据集的处理,搭建网络模型,迭代训练等阶段最后得到一个精准的预测模型。其总体流程图
算法模块总体流程图
一个好的模型开始需要提供一个好的数据集,本系统中采用开源的垃圾分类数据集,该数据集中有着13675张图片。
部分代码