dataloader

最新推荐文章于 2024-07-23 14:36:35 发布

G鲲鹏展翅Y

最新推荐文章于 2024-07-23 14:36:35 发布

阅读量353

点赞数 7

文章标签： python 深度学习机器学习

本文链接：https://blog.csdn.net/qq_44665922/article/details/136596671

版权

import torchvision
from torch.utils.data import DataLoader

#数据集没下载要先下载
# test_data = torchvision.datasets.CIFAR10("./dataset",train=False,transform=torchvision.transforms.ToTensor(),download=True)

test_data = torchvision.datasets.CIFAR10("./dataset",train=False,transform=torchvision.transforms.ToTensor())

test_loader = DataLoader(dataset = test_data,batch_size=4,shuffle=True,num_workers=0,drop_last=False)
img,target = test_data[0]
print(img.shape)
print(target)


for data in test_loader:
    img, target = data
    print(img.shape)
    print(target)