训练一个卷积神经网络,用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.后续步骤 笔记本