trash-classification 基于TensorFlow和Keras的智能垃圾分类系统,用于参加**软件杯双创大赛。 一、 安装运行项目所需的python模块,包括tensorflow | numpy | keras | cv2 二、 train.py用于训练垃圾分类模型,由于训练的数据量过于庞大,因此不一并上传 三、 predict.py用于预测垃圾的类别,首先运行predict.py,然后输入需要预测的文件路径,即可得到结果。 测试结果示例: