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.后续步骤
笔记本