转换不适用于数据集

我是 pytorch 的新手,想了解一些东西。


我正在按如下方式加载 MNIST:


transform_train = transforms.Compose(

    [transforms.ToTensor(),

     transforms.Resize(size, interpolation=2),

     # transforms.Grayscale(num_output_channels=1),

     transforms.RandomHorizontalFlip(p=0.5),

     transforms.Normalize((mean), (std))])



trainset = torchvision.datasets.MNIST(root='./data', train=True,

                                      download=True, transform=transform_train)

trainloader = torch.utils.data.DataLoader(trainset, batch_size=batch_size,

                                          shuffle=True, num_workers=2)

然而,当我探索数据集时,即trainloader.dataset.train_data[0],我得到了一个范围为 [0,255] 且形状为 (28,28) 的张量。


我错过了什么?这是因为转换不是直接应用于数据加载器,而是仅在运行时应用吗?否则我如何探索我的数据?


宝慕林4294392
浏览 145回答 1
1回答
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python