一个简单的垃圾分类模型,VGG16+密集分类器,只针对四种类别:可回收(易拉罐,塑料瓶)、厨余(菜叶,水果皮)、有害(电池)和其他(烟头,碎瓷片)。而且数据集包含的垃圾种类有限,只是一个简单的demo罢了。
.ipynb文件时训练密集分类器的,要用jupyter打开,
.py文件是我自己用来测试密集分类器的,
.h5文件是我训练好的模型直接被.py文件拿去用了。
环境是Windows10,anaconda(python3),TensorFlow2+keras,
数据集网盘链接: https://pan.baidu.com/s/1OdrkJgIrLCg3-ML4LETJSg 提取码: esru