猫狗大战——pytorch+resnet18数据Nettrain+test
cats vs dogs——resnet18数据Nettrain+test数据这是一个在kaggle上的竞赛,原数据提供了25000张图片,本文所使用的数据集来自其中train的8000张,包括4000张猫和4000张狗#将数据导入data_dir = 'E:\\code\\Python\\catanddog\\train'test_dir = 'E:\\code\\Python\\catanddog\\test'class Data(data.Dataset): def __ini