使用卷积神经网络带有垃圾的图片进行分类
本项目是基于深度学习的垃圾分类项目，使用卷积神经网络对垃圾图片进行分类，使用了VGG16模型，使用了迁移学习的思想，将VGG16模型的最后一层进行了修改，使其能够对垃圾图片进行分类。
- Python 3.8
- Pytorch 1.13.1
- torchvision 0.4.2
- numpy 1.19.2
- matplotlib 3.3.2
-
Notifications
You must be signed in to change notification settings - Fork 0
zhaihaojie/GarbageClassification
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
使用卷积神经网络带有垃圾的图片进行分类
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published