<>1 构建图像分类器
训练一个卷积神经网络,用fastai库(建在PyTorch上)将图像分类为纸板,玻璃,金属,纸张,塑料或垃圾。使用了由Gary Thung和Mindy
Yang手动收集的图像数据集。数据集下载地址如下,然后将其移至与笔记本相同的目录中。(注意:需要使用GPU来加速训练。)
https://github.com/garythung/trashnet/blob/master/data/dataset-resized.zip
<https://github.com/garythung/trashnet/blob/master/data/dataset-resized.zip%20>
https://github.com/collindching/Waste-Sorter
<https://github.com/collindching/Waste-Sorter>
笔记本
<https://nbviewer.jupyter.org/github/collindching/Waste-Sorter/blob/master/Waste%20sorter.ipynb>
热门工具 换一换