简介:本文将带你了解如何构建一个基于图像识别的垃圾分类系统,从数据预处理到模型训练和测试,一步步为你揭开神秘面纱。
一、项目架构
整个项目的程序流程图如下,分别为:1.图像数据预处理 2.模型训练 3.模型测试 三个模块
二、图像数据
本文用到的垃圾图片数据主要为以下四类:1.厨余垃圾 2.可回收垃圾 3.其他垃圾 4.有害垃圾。其中每个类别又涉及多个子类别,例如有害垃圾文件夹中又涉及以下多个类别。每类图片数量及图片总数情况如下,共计6038张图像数据。
三、环境介绍
1.环境要求
本项目开发IDE使用的是:Anaconda中的jupyter notebook。由于本项目基于TensorFlow,因此需要以下环境:tensorflow==2.0 pandas scikit-learn numpy OpenCV2 matplotlib。
2.环境安装实例
环境都可以通过pip进行安装。