【图像识别与处理】构建用于垃圾分类的图像分类器

1 构建图像分类器

训练一个卷积神经网络，用fastai库（建在PyTorch上）将图像分类为纸板，玻璃，金属，纸张，塑料或垃圾。使用了由Gary Thung和Mindy Yang手动收集的图像数据集。数据集下载地址如下，然后将其移至与笔记本相同的目录中。（注意：需要使用GPU来加速训练。）
数据集1：
https://download.csdn.net/download/xu1129005165/12456271
数据集2：
https://download.csdn.net/download/xu1129005165/12456273

建模管道：
1.下载并提取图像
2.将图像组织到不同的文件夹中
3.训练模型
4.制作并评估测试预测
5.后续步骤


笔记本