本项目将采用深度学习的方法,搭建一个垃圾分类识别的训练和测试系统,实现智能化垃圾分类。目前,基于ResNet18的垃圾分类识别,在垃圾数据集dataset2,训练集的Accuracy在94%左右,测试集的Accuracy在92%左右。支持常用的分类模型,如resnet18,resnet50,mobilenet_v2,googlenet等
原文地址《垃圾分类数据集+垃圾分类识别训练代码(Pytorch)》:https://panjinquan.blog.csdn.net/article/details/126419147
1