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

1 构建图像分类器 训练一个卷积神经网络,用fastai库(建在PyTorch上)将图像分类为纸板,玻璃,金属,纸张,塑料或垃圾。使用了由Gary Thung和Mindy Yang手动收集的图像数据集。数据集下载地址以下,而后将其移至与笔记本相同的目录中。(注意:须要使用GPU来加速训练。) https://github.com/garythung/trashnet/blob/master/dat
相关文章
相关标签/搜索