File size: 469 Bytes
b83d9ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import torch
from torchvision import datasets, transforms
from torch.utils.data import DataLoader,TensorDataset
transform = transforms.Compose([
transforms.Resize((512, 512)),
transforms.ToTensor(),
])
# 加载训练集和测试集
train_set = datasets.ImageFolder(root='data/cat_vs_dog/train', transform=transform)
test_set = datasets.ImageFolder(root='data/cat_vs_dog/test', transform=transform)
train_data = train_set.imgs
print("train_data:", train_data) |