
垃圾分类数据集及Pytorch训练代码
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目提供了一个详细的垃圾分类的数据集以及使用Python深度学习框架PyTorch进行模型训练的完整代码,旨在促进AI技术在环保领域的应用研究。
垃圾分类识别的训练和测试系统实现了智能化垃圾分类功能。目前使用基于ResNet18模型进行分类,在垃圾数据集dataset2上,训练集准确率约为94%,测试集准确率为约92%。
全部评论 (0)
还没有任何评论哟~


