Skip to content

Latest commit

 

History

History
13 lines (10 loc) · 455 Bytes

README.md

File metadata and controls

13 lines (10 loc) · 455 Bytes

GarbageClassification

使用卷积神经网络带有垃圾的图片进行分类

项目介绍

本项目是基于深度学习的垃圾分类项目,使用卷积神经网络对垃圾图片进行分类,使用了VGG16模型,使用了迁移学习的思想,将VGG16模型的最后一层进行了修改,使其能够对垃圾图片进行分类。

项目环境

  • Python 3.8
  • Pytorch 1.13.1
  • torchvision 0.4.2
  • numpy 1.19.2
  • matplotlib 3.3.2