TypeError: img should be PIL Image. Got class torch.Tensor

最新推荐文章于 2025-04-21 16:14:26 发布

*pprp*

最新推荐文章于 2025-04-21 16:14:26 发布

阅读量1.7w

点赞数 19

分类专栏：深度学习文章标签： pytorch tensor 报错解决

原创文章不要私自转载，自私转载必究责任，如需转载请联系wx:topeijie商谈

本文链接：https://blog.youkuaiyun.com/DD_PP_JJ/article/details/102730050

版权

背景：

在pytorch中使用MNIST数据集，进行可视化，代码如下：

import torch
import torch.nn as nn
import torchvision
from torchvision import datasets, transforms
from torch.autograd import Variable
import matplotlib.pyplot as plt

# part 1: 数据集的获取，torch中提供了数据集的相关API
mnist_train_dataset = datasets.MNIST(root="./data/",
                                      train=True,
                                      download=True,
                                      transform=
                                        transforms.Compose([transforms.ToTensor(), transforms.Normalize(mean=[0.5],std=[0.5]),transforms.Resize((28,28))])
                                    )
                                        
mnist_test_dataset = datasets.MNIST(root="./data/",
                                      train=False,
                                      download=True,
                                      transform = transforms.Compose([transforms.ToTensor(), transforms.Resize((28,28)))
                  )

# part 2: 数据装载， dataloader
data_loader_train = torch.