Невозможно получить изображение в выводе при использовании базы данных MNIST с использованием Pytorch

Question

iter_test = 0

для изображений, меток в test_loader:

iter_test += 1

images = images.view(-1, 28*28)

outputs = model(images)
_, predicted = torch.max(outputs.data, 1)

if iter_test == 1:
    print('PREDICTION')
    print(predicted[0])

    print('LABEL SIZE')
    print(labels.size())

    print('LABEL FOR IMAGE 0')
    print(labels[0])

Я получаю вывод этого кода следующим образом:

ПРОГНОЗ

тензор (7)

РАЗМЕР ЭТИКЕТКИ

torch.Size ([100])

ЭТИКЕТКА ДЛЯ ИЗОБРАЖЕНИЯ 0

тензор(7)

Код работает отлично.Мне было интересно, смогу ли я получить изображение типа «MNIST» в качестве вывода вместе с предсказанием?

Answer 1

Да :).

Вы можете использовать pyplot и показать изображение, загруженное test_loader.

Чек https://www.oreilly.com/learning/not-another-mnist-tutorial-with-tensorflow.

Надеюсь, это поможет!