本代码为python在pytorch上代码,需要建立文件夹,其子文件夹命名为train和val,分别在子文件夹中放入训练和测试数据,改变代码中数据集地址即可实现分类