利用pytorch训练网络---垃圾分类,(resnet18)

数据集包含6种垃圾,分别为cardboard(纸箱),glass(玻璃)、metal(金属)、paper(纸)、plastic(塑料)、其余废品(trash),数据数量较小,仅供学习。python 数据集标准备工做,包括将数据集分为训练集和测试集,制做标签文件。代码utils.pyjson import os import shutil import json path="e://dataset/
相关文章
相关标签/搜索