Я использую набор данных mnist для обучения сети капсул на фоне keras.После тренировки я хочу отобразить изображение из набора данных mnist.Для загрузки изображений используется mnist.load_data ().Данные хранятся как (x_train, y_train), (x_test, y_test).Теперь для визуализации изображения мой код выглядит следующим образом:
img_path = x_test[1]
print(img_path.shape)
plt.imshow(img_path)
plt.show()
Код выводит следующее:
(28, 28, 1)
и ошибка в plt.imshow (img_path) выглядит следующим образом:
TypeError: Invalid dimensions for image data
Как показать изображение в формате png.Помогите!