利用pytorch訓練網絡---垃圾分類,(resnet18)

數據集包含6種垃圾,分別爲cardboard(紙箱),glass(玻璃)、metal(金屬)、paper(紙)、plastic(塑料)、其餘廢品(trash),數據數量較小,僅供學習。python 數據集標準備工做,包括將數據集分爲訓練集和測試集,製做標籤文件。代碼utils.pyjson import os import shutil import json path="e://dataset/
相關文章
相關標籤/搜索