本项目旨在开发一款智能垃圾分类系统,利用人工智能技术实现垃圾自动识别与分类,提高资源回收利用率,助力环保事业。
垃圾分类智能系统主要功能是对上传的垃圾图片进行识别,并返回分类结果(干垃圾、湿垃圾、有害垃圾、可回收垃圾)。该项目采用深度学习图像处理模块,使用的是顺序序贯模型,即最简单的线性结构,从头到尾无分叉。该模型的基本组件包括:
1. `model.add`:添加层;
2. `model.compile`:设置反向传播模式;
3. `model.fit`:设置训练参数并进行训练。
运行环境为Windows 10 + CUDA9.1 + cuDNN7 + TensorFlow-GPU-1.12.0,以及 PyTorch 1.4.0 和 Keras-2.2.4。在项目的开发过程中遇到的许多错误大多与库文件版本不匹配有关,在实现项目前需要确保环境配置正确且各库文件版本对应一致。