trash-classification

基于TensorFlow和Keras的智能垃圾分类系统，用于参加中国软件杯双创大赛。

一、 安装运行项目所需的python模块，包括tensorflow | numpy | keras | cv2

二、 train.py用于训练垃圾分类模型，由于训练的数据量过于庞大，因此不一并上传

三、 predict.py用于预测垃圾的类别，首先运行predict.py，然后输入需要预测的文件路径，即可得到结果。

测试结果示例：