Pytorch: ValueError: 太多维度: 3 > 2. 9/opt/anaconda3/envs/pytorch/lib/python3.7/site-packages/PIL/Image.py in fromarray(obj, mode)

Question

我正在使用 MNIST 数据通过 pytorch 运行我的 python。 我喜欢只训练数字 0 和 1 的部分数据。当我尝试打印第一张图像的大小时，它遇到了这个错误：

值错误：维度太多：3 > 2

我对 Python 很陌生。 如果我不分割训练数据，程序运行良好。 这是代码片段

subset_indices = ((train_data.train_labels == 0) + (train_data.train_labels == 1)).nonzero()
train_loader = torch.utils.data.DataLoader(train_data,batch_size=batch_size, shuffle=False,sampler=SubsetRandomSampler(subset_indices))

Answer 1

该错误是由于您将一个 3 维数组传递到函数Image.fromarray ，该函数可能设置为错误的模式。 您需要确保mode设置为RGB ，使其看起来像Image.fromarray(data, mode='RGB') 。