Изменение формы изображения и печать в Python

Question

Я работаю над данными mnist_fashion. Изображения в mnist_data имеют размер 28x28 пикселей. Чтобы передать их в нейронную сеть (многослойный персептрон), я преобразовал данные в форму (784,).

Кроме того, мне нужно снова изменить его до исходного размера.

Для этого я использовал приведенный ниже код: -

from keras.datasets import fashion_mnist
import numpy as np
import matplotlib.pyplot as plt


(train_imgs,train_lbls), (test_imgs, test_lbls) = fashion_mnist.load_data()
plt.imshow(test_imgs[0].reshape(28,28))

no_of_test_imgs  = test_imgs.shape[0]

test_imgs_trans  = test_imgs.reshape(test_imgs.shape[1]*test_imgs.shape[2], no_of_test_imgs).T

plt.imshow(test_imgs_trans[0].reshape(28,28))

К сожалению, я не получение аналогичного изображения. Я не могу понять, почему это происходит.

ожидаемое изображение:

полученное изображение:

Пожалуйста, помогите мне решить проблему.

Answer 1

обратите внимание, когда вы сводите изображения в test_imgs_trans

(train_imgs,train_lbls), (test_imgs, test_lbls) = tf.keras.datasets.fashion_mnist.load_data()

plt.imshow(test_imgs[0].reshape(28,28))

no_of_test_imgs  = test_imgs.shape[0]

test_imgs_trans  = test_imgs.reshape(no_of_test_imgs, test_imgs.shape[1]*test_imgs.shape[2])

plt.imshow(test_imgs_trans[0].reshape(28,28))