使用华为云垃圾分类数据集训练的垃圾分类网络,可以对40类垃圾分类。网络使用了VGG16作为基础,自己训练了分类器,并且定义了测试接口,可以使用自己的图片测试效果 网络没有过多的调参,只是一个精度不高的draft 数据集整理了网络上能找到的,和保存的模型参数一起放到csdn了,可以免费下载,如果有侵权请联系我删除 https://download.csdn.net/download/liar_liar01/88681800
使用的anaconda环境如下
keras 2.10.0
Keras-Preprocessing 1.1.2
matplotlib 3.7.0
matplotlib-inline 0.1.6
mingw 4.7
numpy 1.24.2
opencv-contrib-python 4.8.0.76
pandas 2.1.0
prompt-toolkit 3.0.39
pygame 2.3.0
python 3.9.16
python-dateutil 2.8.2
python-version 0.0.2
scikit-learn 1.0.2
scipy 1.10.1
tensorboard 2.10.1
tensorboard-data-server 0.6.1
tensorboard-plugin-wit 1.8.1
tensorflow-estimator 2.10.0
tensorflow-gpu 2.10.0
tensorflow-intel 2.11.0
tensorflow-io-gcs-filesystem 0.30.0
termcolor 2.2.0
Theano 1.0.5