深度学习猫狗大战数据集处理7.2

猫狗大战数据集：
链接：https://pan.baidu.com/s/1_qa0u-vLFx1ARmlmJ05R_w
提取码：bgao

import torch
import  torchvision
import os
from torchvision import datasets,transforms
import  time
import matplotlib.pyplot as plt
#载入数据集
data_dir="D:/研究生阶段PPT等文件/DogsVSCats"
data_transform = {x:transforms.Compose([transforms.Scale([64,64]),transforms.ToTensor()])
                  for x in ["train","valid"]}
#transform.Scale  将原始图片的大小统一缩放至64*64
image_datasets={x:datasets.ImageFolder(root=os.path.join(data_dir,x),
                                       transform  = data_transform[x])

                for x in ["train", "valid"]}
#os.path.join将输入参数的两个名字拼接成一个完整的文件目录
dataloader={x:torch.utils.data.DataLoader(dataset=image_datasets[x],
                                          batch_size=16,
                                          shuffle=True)
            for x in ["train","valid&