Неправильная ошибка формы при построении 2D-изображения в python для набора данных языка жестов MNIST, имеющего значения PIXEL AS COLUMNS

Question

У меня есть набор данных языка жестов MNIST со значениями пикселей в виде столбцов.

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

#Training dataset
dfr = pd.read_csv("sign_mnist_train.csv")
X_train_orig = dfr.iloc[:,1:]
Y_train_orig = dfr['label']

#Testing dataset
dfe = pd.read_csv("sign_mnist_test.csv")
X_test_orig = dfe.iloc[:,1:]
Y_test_orig = dfe['label']

#shapes of dataset
print(dfr.shape) #(27455, 785)
print(dfe.shape) #(7172, 785)

#Example of a picture
index = 1
plt.imshow(X_train_orig.iloc[index])



#TypeError: Invalid shape (784,) for image data

Answer 1

Похоже, изображение, которое вы пытаетесь построить, является плоским, соответствующим [B, N], где N - 1x28x28, а B - 27455, что соответствует размеру вашего изображения (27455, 784). Это хорошо, если вы хотите подать его на линейный слой длиной 784 вектора. Чтобы построить это изображение, вы должны изменить его, чтобы оно соответствовало [27455, 1, 28, 28]. Вы можете попробовать это:

image = X_train_orig.iloc[index]
image = np.reshape(image.values, (28, 28))
plt.imshow(image)