Pytorch: ValueError: Слишком много измерений: 3> 2. 9 / opt / anaconda3 / envs / pytorch / lib / python3 .7 / site-packages / PIL / Image.py в fromarray (obj, mode)

Question

Я использую данные MNIST для запуска python с использованием pytorch. Мне нравится тренировать только частичные данные для цифр 0 и 1. Когда я пытаюсь напечатать размер первого изображения, он сталкивается с этой ошибкой:

ValueError: слишком много измерений: 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').