我的数据集路径在D:\wjd这里

好的，你可以按照以下方式修改代码，将数据集路径设置为 `D:\wjd`，并且按照 80% - 20% 的比例划分数据集：

import os
import torchvision.transforms as transforms
import torchvision.datasets as datasets

# 定义数据预处理
transform = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])

# 定义数据集路径
data_dir = 'D:/wjd'

# 将数据集按照 80% - 20% 的比例划分为训练集和验证集
train_dir = os.path.join(data_dir, 'train')
val_dir = os.path.join(data_dir, 'val')
if not os.path.exists(val_dir):
    os.makedirs(val_dir)
    for class_dir in os.listdir(train_dir):
        class_path = os.path.join(train_dir, class_dir)
        if os.path.isdir(class_path):
            os.makedirs(os.path.join(val_dir, class_dir))
            file_list = os.listdir(class_path)
            num_val = int(len(file_list) * 0.2)
            for i in range(num_val):
                file_name = file_list[i]
                src_path = os.path.join(class_path, file_name)
                dst_path = os.path.join(val_dir, class_dir, file_name)
                os.rename(src_path, dst_path)

# 定义数据集
train_data = datasets.ImageFolder(train_dir, transform=transform)
val_data = datasets.ImageFolder(val_dir, transform=transform)

# 定义数据加载器
train_loader = torch.utils.data.DataLoader(train_data, batch_size=32, shuffle=True, num_workers=4)
val_loader = torch.utils.data.DataLoader(val_data, batch_size=32, shuffle=False, num_workers=4)

这里我们只是将 `data_dir` 修改为了 `D:/wjd`，其余的代码和之前的示例完全一样。注意，在划分数据集之前，请确保数据集中的所有图片已经被按照类别存放在不同的文件夹中。