【圖像識別與處理】構建用於垃圾分類的圖像分類器

1 構建圖像分類器 訓練一個卷積神經網絡,用fastai庫(建在PyTorch上)將圖像分類爲紙板,玻璃,金屬,紙張,塑料或垃圾。使用了由Gary Thung和Mindy Yang手動收集的圖像數據集。數據集下載地址以下,而後將其移至與筆記本相同的目錄中。(注意:須要使用GPU來加速訓練。) https://github.com/garythung/trashnet/blob/master/dat
相關文章
相關標籤/搜索